From 6545f59e24e6479a0aa234d945a902a48a1ccf68 Mon Sep 17 00:00:00 2001 From: Galunid Date: Fri, 31 May 2024 08:55:25 +0200 Subject: [PATCH] Handle NotImplementedError in convert-hf-to-gguf --- 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")