Skip to content

Add cpu-only extra for torch cpu dependency#210

Open
voegtlel wants to merge 4 commits intodevelopfrom
feature/cpu_only
Open

Add cpu-only extra for torch cpu dependency#210
voegtlel wants to merge 4 commits intodevelopfrom
feature/cpu_only

Conversation

@voegtlel
Copy link
Collaborator

If installing with extra [cpu-only], the torch dependency will switch to using the cpu-only version to reduce dependency size.

pyproject.toml Outdated
]

[tool.uv.sources]
torch = [
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Did you check that it doesn't also override torch to CPU when not using the "cpu-only" extra?

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Okay, found out that this is actually broken. It would now always use the cpu version :-/ Reverted that change now, added instructions for uv tool.

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.

2 participants