Skip to content

Commit

Permalink
convert.py : add consolidated.safetensors for mixtral 8x22b (ggergano…
Browse files Browse the repository at this point in the history
  • Loading branch information
slaren authored Apr 10, 2024
1 parent 67fac4b commit 65c64dc
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion convert.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 65c64dc

Please sign in to comment.