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

Add --skip to install.py #2314

Closed
wants to merge 2 commits into from
Closed

Add --skip to install.py #2314

wants to merge 2 commits into from

Conversation

xuzhao9
Copy link
Contributor

@xuzhao9 xuzhao9 commented Jun 19, 2024

Use --skip to skip the install of certain models.

Fixes #2308

Test plan:

$ python install.py hf_Bert hf_Bart yolov3 --skip hf_Bert yolov3
checking packages torch, torchvision, torchaudio are installed...OK
running setup for /Users/xzhao9/git/benchmark/torchbenchmark/models/hf_Bart...OK

@facebook-github-bot
Copy link
Contributor

@xuzhao9 has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

@facebook-github-bot
Copy link
Contributor

@xuzhao9 has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

@facebook-github-bot
Copy link
Contributor

@xuzhao9 merged this pull request in 5254910.

@xuzhao9 xuzhao9 deleted the xz9/add-skip-install branch June 20, 2024 18:12
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Add option to skip model for install
3 participants