Skip to content

Commit 65c64dc

Browse files
authored
convert.py : add consolidated.safetensors for mixtral 8x22b (#6587)
1 parent 67fac4b commit 65c64dc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

convert.py

Lines changed: 1 addition & 1 deletion
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)