Hugging Face's logo

Spaces:
Ketengan-Diffusion-Lab
/
Dolphin-Inference
Build error

App Files Community
Ketengan-Diffusion-Lab commited on
Commit
6c5c0c4
·
verified ·
1 Parent(s): f3c7bef

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -1,5 +1,6 @@
1
- torchvision
2
- torch
 
3
  ninja
4
  git+https://github.com/Dao-AILab/flash-attention.git
5
  transformers
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu118
2
+ torch==2.3.1+cu118
3
+ torchvision==0.18.1+cu118
4
  ninja
5
  git+https://github.com/Dao-AILab/flash-attention.git
6
  transformers