Skip to content

Commit a332cd7

Browse files
committed
WIP: fix
1 parent 9114b1a commit a332cd7

2 files changed

Lines changed: 4 additions & 2 deletions

File tree

.github/workflows/continuous-benchmark-transfer.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -127,6 +127,7 @@ jobs:
127127
done
128128
env:
129129
ANSIBLE_HOST_KEY_CHECKING: "False"
130+
ANSIBLE_SSH_ARGS: "-o ServerAliveInterval=30 -o ServerAliveCountMax=10 -o ControlMaster=no"
130131
QDRANT_VERSIONS: ${{ env.QDRANT_VERSIONS }}
131132
DATASET_NAME: ${{ env.DATASET_NAME }}
132133

ansible/playbooks/roles/run-transfer-speed/files/shard_transfer.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -223,8 +223,9 @@ def main():
223223
print(f"Error: {VECTORS_FILE} not found")
224224
sys.exit(1)
225225

226-
vectors = np.load(VECTORS_FILE)
227-
print(f"Loaded {len(vectors):,} vectors ({vectors.shape[1]} dims)")
226+
# Use memory mapping to avoid loading entire file into RAM
227+
vectors = np.load(VECTORS_FILE, mmap_mode='r')
228+
print(f"Loaded {len(vectors):,} vectors ({vectors.shape[1]} dims) [mmap]")
228229

229230
benchmark = TransferBenchmark(QDRANT_URIS, QDRANT_API_KEY)
230231
try:

0 commit comments

Comments
 (0)