Explorar el Código

Fix script for running servers robustly

Aleksandr Borzunov hace 2 años
padre
commit
96033de921
Se han modificado 1 ficheros con 2 adiciones y 2 borrados
  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