diff --git a/src/dstack/_internal/server/background/tasks/process_running_jobs.py b/src/dstack/_internal/server/background/tasks/process_running_jobs.py index 985cbfee8d..aa69f2f335 100644 --- a/src/dstack/_internal/server/background/tasks/process_running_jobs.py +++ b/src/dstack/_internal/server/background/tasks/process_running_jobs.py @@ -1139,7 +1139,7 @@ def _patch_base_image_for_aws_efa( efa_enabled_patterns = [ # TODO: p6-b200 isn't supported yet in gpuhunt r"^p6-b200\.(48xlarge)$", - r"^p5\.(48xlarge)$", + r"^p5\.(4xlarge|48xlarge)$", r"^p5e\.(48xlarge)$", r"^p5en\.(48xlarge)$", r"^p4d\.(24xlarge)$",