From 0515ad93f48df63bbff204eddb0cac75e8585c65 Mon Sep 17 00:00:00 2001 From: Galunid Date: Fri, 31 May 2024 17:42:33 +0200 Subject: [PATCH] convert-hf : Handle NotImplementedError in convert-hf-to-gguf (#7660) --- convert-hf-to-gguf.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py index 9f29cda23..ad071b974 100755 --- a/convert-hf-to-gguf.py +++ b/convert-hf-to-gguf.py @@ -2840,7 +2840,12 @@ def main() -> None: hparams = Model.load_hparams(dir_model) with torch.inference_mode(): - model_class = Model.from_model_architecture(hparams["architectures"][0]) + try: + model_class = Model.from_model_architecture(hparams["architectures"][0]) + except NotImplementedError: + logger.error(f"Model {hparams['architectures'][0]} is not supported") + sys.exit(1) + model_instance = model_class(dir_model, ftype_map[args.outtype], fname_out, args.bigendian, args.use_temp_file, args.no_lazy) logger.info("Set model parameters")