Add script to run servers robustly
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…
Reference in New Issue