change diloco script to not have tail blocking

[samsja]

previously it was not possible to chain two calls to our diloco helper script because the tail was blocking. This pr fix this. Making it easy to run multiple experiments in one go.

Example of what was failing before

#!/bin/bash

for i in {1..3}; do
    ./scripts/simulate_multi_node_diloco.sh 2 2 src/zeroband/train.py @ configs/debug/normal.toml
done

This would stay stuck in the first iteration. This pr fixes it

[Jackmin801]

Oh nice. didnt think of that. I just removed the tail

[samsja]

Oh nice. didnt think of that. I just removed the tail

I did not think about it neither, claude did