Upload requirements.txt
Browse files- requirements.txt +3 -2
requirements.txt
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
huggingface_hub
|
| 2 |
-
torch
|
| 3 |
torchvision
|
| 4 |
-
torchao
|
| 5 |
diffusers
|
| 6 |
peft
|
| 7 |
accelerate
|
|
@@ -12,3 +12,4 @@ Pillow
|
|
| 12 |
sentencepiece
|
| 13 |
protobuf
|
| 14 |
pytorch-lightning
|
|
|
|
|
|
| 1 |
huggingface_hub
|
| 2 |
+
torch==2.4.0
|
| 3 |
torchvision
|
| 4 |
+
torchao==0.9.0
|
| 5 |
diffusers
|
| 6 |
peft
|
| 7 |
accelerate
|
|
|
|
| 12 |
sentencepiece
|
| 13 |
protobuf
|
| 14 |
pytorch-lightning
|
| 15 |
+
triton
|