浏览代码

Fix script for running servers robustly

Aleksandr Borzunov 2 年之前
父节点
当前提交
96033de921
共有 1 个文件被更改,包括 2 次插入2 次删除
  1. 2 2
      src/petals/cli/run_prod_server.sh

+ 2 - 2
src/petals/cli/run_prod_server.sh

@@ -3,10 +3,10 @@ set -x
 
 export HIVEMIND_COLORS=true
 while true; do
+        pkill -f p2p
+        pkill -f run_server
         python -m petals.cli.run_server bigscience/bloom-petals \
                 --block_indices $1 \
                 --torch_dtype bfloat16 --load_in_8bit \
                 --attn_cache_size $2 2>&1 | tee log_`date '+%F_%H:%M:%S'`
-        pkill -f p2p
-        pkill -f run_server
 done