Making tag to upper before converting to enum

harsh97 · harsh97 · commit 31a54e51ec3e · 2024-07-12T00:48:20.000+05:30
diff --git a/ads/aqua/model/model.py b/ads/aqua/model/model.py
@@ -447,7 +447,7 @@ def _process_model(
         inference_containers = AquaContainerConfig.from_container_index_json().inference
 
         model_format = ModelFormat[
-            freeform_tags.get(Tags.MODEL_FORMAT, ModelFormat.SAFETENSORS.value)
+            freeform_tags.get(Tags.MODEL_FORMAT, ModelFormat.SAFETENSORS.value).upper()
         ]
         supported_platform: Set[AquaContainerConfigItem.Platform] = set()
 

Original file line number	Diff line number	Diff line change
`@@ -447,7 +447,7 @@ def _process_model(`
`447`	`447`	`inference_containers = AquaContainerConfig.from_container_index_json().inference`
`448`	`448`
`449`	`449`	`model_format = ModelFormat[`
`450`		`- freeform_tags.get(Tags.MODEL_FORMAT, ModelFormat.SAFETENSORS.value)`
	`450`	`+ freeform_tags.get(Tags.MODEL_FORMAT, ModelFormat.SAFETENSORS.value).upper()`
`451`	`451`	`]`
`452`	`452`	`supported_platform: Set[AquaContainerConfigItem.Platform] = set()`
`453`	`453`