Update convert_hf_to_gguf.py

Co-authored-by: Xuan Son Nguyen <thichthat@gmail.com>
This commit is contained in:
Brian 2024-07-13 20:42:55 +10:00 committed by brian khuu
parent ad217d7249
commit 60278e4f4d

View file

@ -3624,7 +3624,7 @@ def main() -> None:
try: try:
model_class = Model.from_model_architecture(model_architecture) model_class = Model.from_model_architecture(model_architecture)
except NotImplementedError: except NotImplementedError:
logger.error(f"Model {hparams['architectures'][0]} is not supported") logger.error(f"Model {model_architecture} is not supported")
sys.exit(1) sys.exit(1)
model_instance = model_class(dir_model=dir_model, ftype=output_type, fname_out=fname_out, model_instance = model_class(dir_model=dir_model, ftype=output_type, fname_out=fname_out,