Skip to content

Commit 241dfee

Browse files
committed
fixed mmd
1 parent 94123be commit 241dfee

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

ads/aqua/common/entities.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -287,7 +287,7 @@ class AquaMultiModelRef(Serializable):
287287
description="Environment variables to override during container startup.",
288288
)
289289
params: Optional[dict] = Field(
290-
default_factory=dict,
290+
default=None,
291291
description=(
292292
"Framework-specific startup parameters required by the container runtime. "
293293
"For example, vLLM models may use flags like `--tensor-parallel-size`, `--enforce-eager`, etc."

ads/aqua/modeldeployment/model_group_config.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -178,9 +178,10 @@ def _merge_gpu_count_params(
178178

179179
final_model_params = user_params
180180
params_found = False
181+
user_explicitly_cleared = model.params is not None and not model.params
181182

182-
# If user DID NOT provide specific params (None or Empty), we look for defaults
183-
if not user_params:
183+
# Only load defaults if user didn't provide params AND didn't explicitly clear them
184+
if not user_params and not user_explicitly_cleared:
184185
for item in deployment_config.multi_model_deployment:
185186
if (
186187
model.gpu_count

0 commit comments

Comments
 (0)