Update convert_llama.py

intel · May 28, 2024 · 235cd03 · 235cd03
1 parent b847728
commit 235cd03
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neural_speed/convert/convert_llama.py b/neural_speed/convert/convert_llama.py
@@ -1357,7 +1357,7 @@ def load_some_model(path: Path) -> ModelPlus:
     if path.is_dir():
         # Check if it's a set of safetensors files first
         files = list(path.glob("model-00001-of-*.safetensors"))
-        if not files
+        if not files:
             files = list(path.glob("model*.safetensors")) # for only one safetensor
         if not files:
             # Try the PyTorch patterns too, with lower priority