Hugging Face's logo

Spaces:
unik-style
/
unik-ml
Runtime error

App Files Community
3
RoniFinTech commited on
Commit
98343d1
•
1 Parent(s): 67c072d

gpu

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu118
2
  fastapi==0.100.1
3
  pydantic==2.1.1
4
  pylint==2.17.5
@@ -7,4 +7,5 @@ torch==2.0.1
7
  transformers==4.31.0
8
  accelerate==0.21.0
9
  diffusers==0.19.3
10
- torchvision==0.15.2
 
 
1
+ # --extra-index-url https://download.pytorch.org/whl/cu118
2
  fastapi==0.100.1
3
  pydantic==2.1.1
4
  pylint==2.17.5
 
7
  transformers==4.31.0
8
  accelerate==0.21.0
9
  diffusers==0.19.3
10
+ torchvision==0.15.2
11
+ safetensors==0.3.1