Example88 commited on
Commit
cbe853c
·
1 Parent(s): 58867bf

Fix: pin torchvision to 0.14.1 for functional_tensor

Browse files
Files changed (1) hide show
  1. requirements.txt +10 -9
requirements.txt CHANGED
@@ -1,14 +1,15 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu118
 
 
 
 
2
 
3
- # واجهة المستخدم
4
- gradio
5
 
6
- # نماذج الذكاء الاصطناعي
7
  realesrgan
8
  gfpgan
9
 
10
- # أساسيات
11
- torch
12
- torchvision
13
- torchaudio
14
- pillow
 
1
+ # Core ML dependencies
2
+ torch==2.1.2
3
+ torchvision==0.14.1
4
+ # torchvision>=0.13.1 includes functional_tensor module
5
+ torchaudio==2.1.2
6
 
7
+ # Gradio UI
8
+ gradio>=4.29
9
 
10
+ # Restoration & Upscaling
11
  realesrgan
12
  gfpgan
13
 
14
+ # Image handling
15
+ pillow