Skip to content

Pin PyTorch version to 2.1#5056

Merged
oobabooga merged 3 commits into
devfrom
pin-pytorch
Jan 5, 2024
Merged

Pin PyTorch version to 2.1#5056
oobabooga merged 3 commits into
devfrom
pin-pytorch

Conversation

@oobabooga
Copy link
Copy Markdown
Owner

Without this change the one-click installer may break on January 11th when PyTorch 2.2 gets released. The flash-attention wheels assume PyTorch 2.1.

https://dev-discuss.pytorch.org/t/pytorch-release-2-2-0-call-for-features/1629

@oobabooga oobabooga mentioned this pull request Dec 23, 2023
1 task
@oobabooga oobabooga merged commit 3d854ee into dev Jan 5, 2024
@oobabooga oobabooga deleted the pin-pytorch branch January 5, 2024 02:56
PoetOnTheRun pushed a commit to PoetOnTheRun/text-generation-webui that referenced this pull request Feb 22, 2024
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.

1 participant