diff --git a/llms/llama/convert.py b/llms/llama/convert.py index ec53ba64..dae337ee 100644 --- a/llms/llama/convert.py +++ b/llms/llama/convert.py @@ -63,7 +63,7 @@ def tiny_llama(model_path): except ImportError: print("The transformers package must be installed for this model conversion:") print("pip install transformers") - exit(0) + exit(1) model = transformers.AutoModelForCausalLM.from_pretrained( str(model_path)