File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff 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 ,
You can’t perform that action at this time.
0 commit comments