Add script to run servers robustly

pull/104/head
Aleksandr Borzunov 2 years ago
parent 088713912d
commit 85cf32d2a4

@ -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
Loading…
Cancel
Save