Skip to content

Commit 4b8ea8a

Browse files
committed
Fixes loading default params for the AQUA model.
1 parent f38230a commit 4b8ea8a

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

ads/aqua/common/enums.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,7 @@ class InferenceContainerTypeFamily(ExtendedEnum):
5858
AQUA_VLLM_LLAMA4_CONTAINER_FAMILY = "odsc-vllm-serving-llama4"
5959
AQUA_TGI_CONTAINER_FAMILY = "odsc-tgi-serving"
6060
AQUA_LLAMA_CPP_CONTAINER_FAMILY = "odsc-llama-cpp-serving"
61+
AQUA_VLLM_OPENAI_CONTAINER_FAMILY = "odsc-vllm-serving-openai"
6162

6263

6364
class CustomInferenceContainerTypeFamily(ExtendedEnum):

ads/aqua/modeldeployment/deployment.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1160,10 +1160,7 @@ def get_deployment_default_params(
11601160
f"{AQUA_DEPLOYMENT_CONTAINER_METADATA_NAME} key is not available in the custom metadata field for model {model_id}."
11611161
)
11621162

1163-
if (
1164-
container_type_key
1165-
and container_type_key in InferenceContainerTypeFamily.values()
1166-
):
1163+
if container_type_key:
11671164
deployment_config = self.get_deployment_config(model_id)
11681165

11691166
instance_shape_config = deployment_config.configuration.get(

0 commit comments

Comments
 (0)