fix(requirements)
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
@@ -7,4 +7,5 @@ tensorboard
|
|
7 |
Jinja2
|
8 |
peft==0.7.1
|
9 |
diffusers==0.24.0
|
10 |
-
|
|
|
|
7 |
Jinja2
|
8 |
peft==0.7.1
|
9 |
diffusers==0.24.0
|
10 |
+
--find-links https://download.pytorch.org/whl/torch_stable.html
|
11 |
+
torch==2.1.2+cu121
|