mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-08-09 18:36:38 +08:00
use non-zero exit code on error (#177)
This commit is contained in:
parent
092e87211e
commit
848f118ac5
@ -63,7 +63,7 @@ def tiny_llama(model_path):
|
|||||||
except ImportError:
|
except ImportError:
|
||||||
print("The transformers package must be installed for this model conversion:")
|
print("The transformers package must be installed for this model conversion:")
|
||||||
print("pip install transformers")
|
print("pip install transformers")
|
||||||
exit(0)
|
exit(1)
|
||||||
|
|
||||||
model = transformers.AutoModelForCausalLM.from_pretrained(
|
model = transformers.AutoModelForCausalLM.from_pretrained(
|
||||||
str(model_path)
|
str(model_path)
|
||||||
|
Loading…
Reference in New Issue
Block a user