@@ -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