From 0479bafbf3e4e85fd3a1a1e084f692305cb03a1d Mon Sep 17 00:00:00 2001 From: YAN Wenkun Date: Thu, 31 Oct 2024 05:30:50 +0800 Subject: [PATCH] Upgrade to xformers 0.0.28.post3, torch 2.5.1 --- stage1.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/stage1.sh b/stage1.sh index 429d1cc..e0c5f34 100644 --- a/stage1.sh +++ b/stage1.sh @@ -42,7 +42,7 @@ curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py --upgrade pip wheel setuptools Cython cmake ./python.exe -s -m pip install \ - xformers==0.0.28.post2 torch==2.5.0 torchvision torchaudio \ + xformers==0.0.28.post3 torch==2.5.1 torchvision torchaudio \ --index-url https://download.pytorch.org/whl/cu124 \ --extra-index-url https://pypi.org/simple