We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 9136b8c commit aa65ad8Copy full SHA for aa65ad8
convert.py
@@ -1350,7 +1350,7 @@ def load_some_model(path: Path) -> ModelPlus:
1350
# Be extra-friendly and accept either a file or a directory:
1351
if path.is_dir():
1352
# Check if it's a set of safetensors files first
1353
- globs = ["model-00001-of-*.safetensors", "model.safetensors"]
+ globs = ["model-00001-of-*.safetensors", "model.safetensors", "consolidated.safetensors"]
1354
files = [file for glob in globs for file in path.glob(glob)]
1355
if not files:
1356
# Try the PyTorch patterns too, with lower priority
0 commit comments