Pārlūkot izejas kodu

Add script to run servers robustly

Aleksandr Borzunov 2 gadi atpakaļ
vecāks
revīzija
85cf32d2a4
1 mainītis faili ar 12 papildinājumiem un 0 dzēšanām
  1. 12 0
      src/petals/cli/run_prod_server.sh

+ 12 - 0
src/petals/cli/run_prod_server.sh

@@ -0,0 +1,12 @@
+#!/bin/bash
+set -x
+
+export HIVEMIND_COLORS=true
+while true; do
+        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