common.py 502 B

1234567891011121314151617181920
  1. #!/usr/bin/env python3
  2. import os
  3. import sys
  4. def get_model_name_from_env_path(env_path_name):
  5. model_path = os.getenv(env_path_name)
  6. if not model_path:
  7. print(f"Error: {env_path_name} environment variable not set")
  8. sys.exit(1)
  9. if not os.path.exists(model_path):
  10. print(f"Error: Model file not found: {model_path}")
  11. sys.exit(1)
  12. name = os.path.basename(os.path.normpath(model_path))
  13. if name.endswith(".gguf"):
  14. name = name[:-5]
  15. return name