|
|
@@ -62,7 +62,7 @@ jobs:
|
|
|
- name: Test
|
|
|
run: |
|
|
|
|
|
|
- hivemind-dht --host_maddrs /ip4/127.0.0.1/tcp/0 &> dht.log &
|
|
|
+ hivemind-dht --host_maddrs /ip4/127.0.0.1/tcp/31337 &> dht.log &
|
|
|
INITIAL_PID=$$
|
|
|
sleep 5
|
|
|
export INITIAL_PEERS=$(python -c "with open('dht.log') as f: print(f.readlines()[1].split()[-1])" )
|
|
|
@@ -70,7 +70,7 @@ jobs:
|
|
|
cat dht.log
|
|
|
|
|
|
python -m cli.run_server --converted_model_name_or_path bloom-testing/test-bloomd-350m --block_indices 0:12 \
|
|
|
- --torch_dtype float32 --host_maddrs /ip4/127.0.0.1/tcp/0 --initial_peers $INITIAL_PEERS --throughput 1 &
|
|
|
+ --torch_dtype float32 --initial_peers $INITIAL_PEERS --throughput 1 &
|
|
|
SERVER1_PID=$$
|
|
|
|
|
|
sleep 60 # wait for server to download layers
|
justheuristic