浏览代码

Add script to run servers robustly

Aleksandr Borzunov 2 年之前
父节点
当前提交
85cf32d2a4
共有 1 个文件被更改,包括 12 次插入0 次删除
  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