Browse Source

Add script to run servers robustly

Aleksandr Borzunov 2 năm trước cách đây
mục cha
commit
85cf32d2a4
1 tập tin đã thay đổi với 12 bổ sung0 xóa
  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