Aklavya commited on
Commit
20d431f
·
verified ·
1 Parent(s): a37cd75

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +16 -14
requirements.txt CHANGED
@@ -1,14 +1,16 @@
1
- diffusers
2
- torch~=2.0.1
3
- torchvision
4
- pipeline
5
- transformers
6
- accelerate
7
- safetensors
8
- sentencepiece
9
- spaces
10
- peft
11
- gradio~=5.8.0
12
- numpy~=1.26.0
13
- xformers
14
- python-dotenv
 
 
 
1
+ torch==2.4.1
2
+ torchvision==0.19.1
3
+ # don't install xformers on CPU
4
+
5
+ diffusers==0.30.3
6
+ transformers==4.45.2
7
+ accelerate==1.0.1
8
+ safetensors==0.4.5
9
+
10
+ gradio==5.8.0
11
+ numpy==1.26.4
12
+ Pillow>=10.3
13
+ spaces==0.30.1
14
+ python-dotenv>=1.0.1
15
+ sentencepiece>=0.2.0
16
+ peft>=0.13.0