From d285fb6558411efcd105460f49b1e4880553f1bd Mon Sep 17 00:00:00 2001 From: slaren Date: Wed, 10 Apr 2024 14:39:43 +0200 Subject: [PATCH] convert.py : add consolidated.safetensors for mixtral 8x22b --- convert.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/convert.py b/convert.py index e860ac89f..24df0a4d8 100755 --- a/convert.py +++ b/convert.py @@ -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