Rifaa commited on
Commit
3ff4a0e
·
verified ·
1 Parent(s): d7e5c31

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -3
requirements.txt CHANGED
@@ -1,7 +1,9 @@
1
- gradio
2
  torch
3
- diffusers
4
  jinja2
5
  numpy
6
- opencv-python
 
7
  transformers
 
 
 
 
 
1
  torch
 
2
  jinja2
3
  numpy
4
+ torch==2.0.1+cu117 # Adjust based on the GPU's CUDA version
5
+ diffusers>=0.12.0
6
  transformers
7
+ accelerate
8
+ opencv-python
9
+ gradio