浏览代码

more debugprints

justheuristic 3 年之前
父节点
当前提交
73a79cfd30
共有 1 个文件被更改,包括 2 次插入1 次删除
  1. 2 1
      .github/workflows/run-tests.yaml

+ 2 - 1
.github/workflows/run-tests.yaml

@@ -63,9 +63,10 @@ jobs:
           sleep 3
           export INITIAL_PEERS=$(python -c "with open('dht.log') as f: print(f.readlines()[1].split()[-1])" )
           echo "Initial peer: ${INITIAL_PEERS}"
+          cat dht.log
           
           python -m cli.run_server --converted_model_name_or_path bloom-testing/test-bloomd-350m \
-            --block_indices 0:12 --torch_dtype float32 --initial_peers $INITIAL_PEERS &
+            --block_indices 0:12 --torch_dtype float32 --initial_peers $INITIAL_PEERS --throughput 1 &
           SERVER1_PID=$$
           
           sleep 60  # wait for server to download layers