vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix too relaxed model glob (breaking multifile)

Browse source
This commit is contained in:
alex 2023-05-04 00:16:12 +02:00
parent b59c371035
commit 2b7cf9f32b
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
convert.py
View file

@ -1039,7 +1039,12 @@ def load_some_model(path: Path) -> ModelPlus:
'''Load a model of any supported format.''' '''Load a model of any supported format.'''
# Be extra-friendly and accept either a file or a directory: # Be extra-friendly and accept either a file or a directory:
if path.is_dir(): if path.is_dir():
globs = ["consolidated.00.pth", "pytorch_model*.bin", "*.pt"] globs = [
"consolidated.00.pth",
"pytorch_model.bin",
"pytorch_model-00001-of-*.bin",
"*.pt"
]
files = [file for glob in globs for file in path.glob(glob)] files = [file for glob in globs for file in path.glob(glob)]
if not files: if not files:
# Try GGML too, but with lower priority, since if both a non-GGML # Try GGML too, but with lower priority, since if both a non-GGML