From 319e6a103b3030dd4076d33101f1c94f28208e88 Mon Sep 17 00:00:00 2001 From: TheBloke Date: Mon, 15 May 2023 22:11:46 +0100 Subject: [PATCH] Support models in a single pytorch_model.bin --- convert.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/convert.py b/convert.py index 8f4f0399e..7f64f9577 100644 --- a/convert.py +++ b/convert.py @@ -1055,7 +1055,7 @@ def load_some_model(path: Path) -> ModelPlus: files = list(path.glob("model-00001-of-*.safetensors")) if not files: # Try the PyTorch patterns too, with lower priority - globs = ["consolidated.00.pth", "pytorch_model-00001-of-*.bin", "*.pt"] + globs = ["consolidated.00.pth", "pytorch_model-00001-of-*.bin", "*.pt", "pytorch_model.bin" ] files = [file for glob in globs for file in path.glob(glob)] if not files: # Try GGML too, but with lower priority, since if both a non-GGML