Skip to content

Conversation

lianakoleva
Copy link
Collaborator

Addresses #2122 - B200s will be supported out of the box.

@lianakoleva lianakoleva changed the title ci: bump pytorch to >= 2.7 feat: bump pytorch to >= 2.7 Sep 11, 2025
@lianakoleva lianakoleva changed the title feat: bump pytorch to >= 2.7 chore: bump pytorch to >= 2.7 Sep 11, 2025
Copy link
Collaborator

@t-vi t-vi left a comment

Choose a reason for hiding this comment

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

Thank you @lianakoleva

@t-vi t-vi enabled auto-merge (squash) September 11, 2025 17:13
Copy link
Collaborator

@Borda Borda left a comment

Choose a reason for hiding this comment

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

Why is it needed? 🤔
since it installs 2.8 anyway
and this means no one can run it with 2.6 anymore...

@lianakoleva
Copy link
Collaborator Author

i think there are different defaults that will be installed depending on user OS? if we require 2.7 then 2.7 will be installed

@Borda
Copy link
Collaborator

Borda commented Sep 11, 2025

So bump it if you need it for the target env, don't force everyone else...
pip install litgpt torch>=2.7

@lianakoleva
Copy link
Collaborator Author

I don't know of a straightforward way to detect B200 in pyproject.toml, do you have an idea of how to make it work out of the box or were you thinking including it as an extra (e.g., pip install litgpt[b200]) and adding documentation?

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.

6 participants