From c7bae1e125f609ab8ef937eef51fd0cd7fc5967c Mon Sep 17 00:00:00 2001
From: afrideva <95653597+afrideva@users.noreply.github.com>
Date: Sat, 11 Nov 2023 20:20:07 -0800
Subject: [PATCH] Check for single-file safetensors model

---
 convert.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/convert.py b/convert.py
index 51233416c..3d6216f1d 100755
--- a/convert.py
+++ b/convert.py
@@ -1036,7 +1036,8 @@ def load_some_model(path: Path) -> ModelPlus:
     # Be extra-friendly and accept either a file or a directory:
     if path.is_dir():
         # Check if it's a set of safetensors files first
-        files = list(path.glob("model-00001-of-*.safetensors"))
+        globs = ["model-00001-of-*.safetensors", "model.safetensors"]
+        files = [file for glob in globs for file in path.glob(glob)]
         if not files:
             # Try the PyTorch patterns too, with lower priority
             globs = ["consolidated.00.pth", "pytorch_model-00001-of-*.bin", "*.pt", "pytorch_model.bin"]