upload_model.py 429 B

1234567891011121314151617181920212223
  1. """Upload the model files to cloud storage (s3 or gcs).
  2. Example usage:
  3. ```bash
  4. python -m ray.llm.utils.upload_model \
  5. --model-source facebook/opt-350m \
  6. --bucket-uri gs://my-bucket/path/to/facebook-opt-350m
  7. ```
  8. """
  9. import typer
  10. from ray.llm._internal.common.utils.upload_utils import (
  11. upload_model_cli as _upload_model_cli,
  12. )
  13. def main():
  14. typer.run(_upload_model_cli)
  15. if __name__ == "__main__":
  16. main()