Skip to content

Commit 04ee73b

Browse files
authored
Merge pull request #172 from VectorInstitute/f/random-ports
Fix: Randomize port search to avoid port collision during model loading
2 parents 3e1a3ee + 8ac4aaa commit 04ee73b

File tree

2 files changed

+14
-6
lines changed

2 files changed

+14
-6
lines changed

vec_inf/config/models.yaml

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1059,12 +1059,11 @@ models:
10591059
model_family: gpt-oss
10601060
model_variant: 120b
10611061
model_type: LLM
1062-
gpus_per_node: 4
1063-
num_nodes: 2
1062+
gpus_per_node: 2
1063+
num_nodes: 1
10641064
vocab_size: 201088
10651065
time: 08:00:00
10661066
resource_type: l40s
10671067
vllm_args:
1068-
--tensor-parallel-size: 4
1069-
--pipeline-parallel-size: 2
1070-
--max-model-len: 40960
1068+
--tensor-parallel-size: 2
1069+
--max-model-len: 32768

vec_inf/find_port.sh

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,16 @@ find_available_port() {
2828
local base_port=$2
2929
local max_port=$3
3030

31-
for ((port=base_port; port<=max_port; port++)); do
31+
# Generate shuffled list of ports; fallback to sequential if shuf not present
32+
if command -v shuf >/dev/null 2>&1; then
33+
local port_list
34+
port_list=$(shuf -i "${base_port}-${max_port}")
35+
else
36+
local port_list
37+
port_list=$(seq $base_port $max_port)
38+
fi
39+
40+
for port in $port_list; do
3241
if is_port_available $ip $port; then
3342
echo $port
3443
return

0 commit comments

Comments
 (0)