Skip to content

Commit 9d0f257

Browse files
authored
Needuv/foundation model deployment failures (#33014)
* stop llama getting treated as an azureml model * fix sku selection azureml registry
1 parent 1010508 commit 9d0f257

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

sdk/ai/azure-ai-resources/azure/ai/resources/operations/_deployment_operations.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -158,8 +158,12 @@ def create_or_update(self, deployment: Deployment) -> Any:
158158
default_instance_type, deployment, allowed_instance_types=allowed_instance_types
159159
)
160160

161-
if "registries/azureml" in model_details.id:
161+
if "registries/azureml/" in model_details.id:
162162
default_instance_type = model_details.properties["inference-recommended-sku"]
163+
allowed_instance_types = []
164+
if "," in default_instance_type:
165+
allowed_instance_types = model_details.properties["inference-recommended-sku"].split(",")
166+
default_instance_type = allowed_instance_types[0]
163167
min_sku_spec = model_details.properties["inference-min-sku-spec"].split("|")
164168
self._check_default_instance_type_and_populate(
165169
default_instance_type, deployment, min_sku_spec=min_sku_spec

0 commit comments

Comments
 (0)