Skip to content

Commit dc7b47d

Browse files
committed
Updated pr.
1 parent 9810f0c commit dc7b47d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ads/aqua/model/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -933,9 +933,9 @@ def _validate_model(
933933
# now as we know that at least one type of model files exist, validate the content of oss path.
934934
# for safetensors, we check if config.json files exist, and for gguf format we check if files with
935935
# gguf extension exist.
936-
if ModelFormat.SAFETENSORS in model_formats and ModelFormat.GGUF in model_formats:
936+
if {ModelFormat.SAFETENSORS, ModelFormat.GGUF}.issubset(set(model_formats)):
937937
if (
938-
import_model_details.inference_container == InferenceContainerTypeFamily.AQUA_LLAMA_CPP_CONTAINER_FAMILY
938+
import_model_details.inference_container.lower() == InferenceContainerTypeFamily.AQUA_LLAMA_CPP_CONTAINER_FAMILY
939939
):
940940
self._validate_gguf_format(
941941
import_model_details=import_model_details,

0 commit comments

Comments
 (0)