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

py : use cpu-only torch in requirements.txt #8335

Merged
merged 1 commit into from
Jul 7, 2024

Conversation

compilade
Copy link
Collaborator

Since #5745 makes the toplevel pyproject.toml depend on cpu-only torch from https://download.pytorch.org/whl/cpu, I think it might be good to also make the requirements.txt files use the cpu-only torch wheels.

This uses --extra-index-url, which has some pitfalls but using it to get platform-specific wheels seems like an intended use-case.

This reduces the network usage of the check-requirements.sh script; the check on GitHub seems to go 3x faster than before.

@github-actions github-actions bot added examples python python script changes labels Jul 6, 2024
@compilade compilade added the Review Complexity : Low Trivial changes to code that most beginner devs (or those who want a break) can tackle. e.g. UI fix label Jul 6, 2024
@ggerganov ggerganov merged commit d39130a into master Jul 7, 2024
10 checks passed
arthw pushed a commit to arthw/llama.cpp that referenced this pull request Jul 13, 2024
arthw pushed a commit to arthw/llama.cpp that referenced this pull request Jul 13, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
examples python python script changes Review Complexity : Low Trivial changes to code that most beginner devs (or those who want a break) can tackle. e.g. UI fix
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants