Skip to content

Commit aa65ad8

Browse files
slarentybalex
authored andcommitted
convert.py : add consolidated.safetensors for mixtral 8x22b (ggml-org#6587)
1 parent 9136b8c commit aa65ad8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

convert.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1350,7 +1350,7 @@ def load_some_model(path: Path) -> ModelPlus:
13501350
# Be extra-friendly and accept either a file or a directory:
13511351
if path.is_dir():
13521352
# Check if it's a set of safetensors files first
1353-
globs = ["model-00001-of-*.safetensors", "model.safetensors"]
1353+
globs = ["model-00001-of-*.safetensors", "model.safetensors", "consolidated.safetensors"]
13541354
files = [file for glob in globs for file in path.glob(glob)]
13551355
if not files:
13561356
# Try the PyTorch patterns too, with lower priority

0 commit comments

Comments
 (0)