Spaces:
Sleeping
Sleeping
arjunanand13
commited on
Update requirements.txt
Browse files- requirements.txt +1 -0
requirements.txt
CHANGED
@@ -2,4 +2,5 @@ torch @ https://download.pytorch.org/whl/cu121_full/torch-2.4.1%2Bcu121-cp310-cp
|
|
2 |
torchvision @ https://download.pytorch.org/whl/cu121_full/torchvision-0.19.1%2Bcu121-cp310-cp310-linux_x86_64.whl#sha256=b8cc4bf381b75522995b601e07a1b433b5fd925dc3e34a7fa6cd22f449d65379
|
3 |
einops
|
4 |
timm
|
|
|
5 |
#flash_attn
|
|
|
2 |
torchvision @ https://download.pytorch.org/whl/cu121_full/torchvision-0.19.1%2Bcu121-cp310-cp310-linux_x86_64.whl#sha256=b8cc4bf381b75522995b601e07a1b433b5fd925dc3e34a7fa6cd22f449d65379
|
3 |
einops
|
4 |
timm
|
5 |
+
transformers
|
6 |
#flash_attn
|