dream2589632147 commited on
Commit
b9fa8d7
·
verified ·
1 Parent(s): 44a46c9

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -4
requirements.txt CHANGED
@@ -1,8 +1,8 @@
1
- --extra-index-url https://download.pytorch.org/whl/cpu
2
- torch==2.1.0
3
- torchvision==0.16.0
4
  transformers==4.44.0
5
- diffusers==0.29.0
6
  accelerate==0.31.0
7
  safetensors
8
  sentencepiece
@@ -18,3 +18,5 @@ gradio==4.44.0
18
  spaces==0.20.0
19
  imageio-ffmpeg
20
  opencv-python-headless
 
 
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu121
2
+ torch==2.4.0
3
+ torchvision==0.19.0
4
  transformers==4.44.0
5
+ diffusers>=0.35.1
6
  accelerate==0.31.0
7
  safetensors
8
  sentencepiece
 
18
  spaces==0.20.0
19
  imageio-ffmpeg
20
  opencv-python-headless
21
+ torchao # للكمية
22
+ aoti # للـ AoT