Skip to content

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

Merged
ggerganov merged 1 commit intomasterfrom
compilade/requirements-cpu-torch
Jul 7, 2024
Merged

py : use cpu-only torch in requirements.txt#8335
ggerganov merged 1 commit intomasterfrom
compilade/requirements-cpu-torch

Conversation

@compilade
Copy link
Copy Markdown
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
arthw pushed a commit to arthw/llama.cpp that referenced this pull request Jul 13, 2024
Seunghhon pushed a commit to Seunghhon/llama.cpp that referenced this pull request Apr 26, 2026
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