Update requirements.txt
Browse files- requirements.txt +2 -2
requirements.txt
CHANGED
@@ -3,7 +3,7 @@ tqdm
|
|
3 |
transformers
|
4 |
diffusers
|
5 |
safetensors
|
6 |
-
torchvision
|
7 |
sentencepiece
|
8 |
einops
|
9 |
-
bitsandbytes
|
|
|
3 |
transformers
|
4 |
diffusers
|
5 |
safetensors
|
6 |
+
torchvision==0.18.1
|
7 |
sentencepiece
|
8 |
einops
|
9 |
+
bitsandbytes>=0.43.3
|