From: slaren <redacted>
Date: Wed, 10 Apr 2024 13:23:12 +0000 (+0200)
Subject: convert.py : add consolidated.safetensors for mixtral 8x22b (#6587)
X-Git-Tag: upstream/0.0.4488~1843
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=65c64dc36f9bca5b3f100614cdd02bf12d6b3e49;p=pkg%2Fggml%2Fsources%2Fllama.cpp

convert.py : add consolidated.safetensors for mixtral 8x22b (#6587)
---

diff --git a/convert.py b/convert.py
index e860ac89..24df0a4d 100755
--- a/convert.py
+++ b/convert.py
@@ -1350,7 +1350,7 @@ 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
-        globs = ["model-00001-of-*.safetensors", "model.safetensors"]
+        globs = ["model-00001-of-*.safetensors", "model.safetensors", "consolidated.safetensors"]
         files = [file for glob in globs for file in path.glob(glob)]
         if not files:
             # Try the PyTorch patterns too, with lower priority