Hugging Face's logo

Spaces:
CamiloVega
/
News_AI
Runtime error

App Files Community
CamiloVega commited on
Commit
816dc16
·
verified ·
1 Parent(s): 2d4c853

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -4
requirements.txt CHANGED
@@ -1,8 +1,9 @@
1
  transformers==4.36.2
2
- --extra-index-url https://download.pytorch.org/whl/cpu
3
- torch==2.4.0+cpu
4
- torchvision==0.15.2+cpu
5
- torchaudio==2.4.0+cpu
 
6
  accelerate==0.27.2
7
  gradio==4.19.2
8
  huggingface-hub==0.20.3
 
1
  transformers==4.36.2
2
+ --extra-index-url https://download.pytorch.org/whl/cu121 # Usa CUDA 12.1
3
+ torch==2.4.0
4
+ torchvision==0.15.2
5
+ torchaudio==2.4.0
6
+ accelerate==0.27.2
7
  accelerate==0.27.2
8
  gradio==4.19.2
9
  huggingface-hub==0.20.3