File tree Expand file tree Collapse file tree
ansible/playbooks/roles/run-transfer-speed/files Expand file tree Collapse file tree Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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 :
You can’t perform that action at this time.
0 commit comments