Skip to content

Commit

Permalink
Do not autoscale deepem-gpu cluster
Browse files Browse the repository at this point in the history
It does not make sense to auto scale the cluster since the size is fixed
  • Loading branch information
ranlu committed Jan 17, 2025
1 parent 2c59d7e commit bc5cd72
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion dags/cluster_dag.py
Original file line number Diff line number Diff line change
Expand Up @@ -107,7 +107,7 @@ def cluster_control():
cluster_api.resize_instance_group(cluster_info[key], 0)
continue

if num_workers != target_sizes[key]:
if num_workers != target_sizes[key] and key != "deepem-gpu":
target_sizes[key] = max(num_workers, 1)

if stable and requested_size < target_sizes[key]:
Expand Down

0 comments on commit bc5cd72

Please sign in to comment.