Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Avoid downloading HF models in CI pipelines #1263

Merged
merged 14 commits into from
Feb 20, 2025

Conversation

baijumeswani
Copy link
Collaborator

@baijumeswani baijumeswani commented Feb 19, 2025

This pull-request makes changes to the tests so that the HuggingFace models are not downloaded if these are already available on the machine under some directory. The directory it expects the models to be located in is

  • Windows: C:\data\models\ortgenai\pytorch
  • Linux: /data/ortdenai/pytorch

If the models are not found in that location, the test script will download from huggingface_hub.

The models are not made available on macos as we don't have a way to add those models to the macos vm yet.

This pull-request also changes how many models are converted in the CI pipeline to onnx to being only phi-2, and qwen-2.5 for now. I'll create a separate pull-request to enable more models for the nightly build.

@RyanUnderhill
Copy link
Member

:shipit:

@baijumeswani baijumeswani merged commit fba80c1 into main Feb 20, 2025
14 checks passed
@baijumeswani baijumeswani deleted the baijumeswani/avoid-downloading-models branch February 20, 2025 23:42
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants