VcRlAgent commited on
Commit
6e07159
·
1 Parent(s): d149111

Init Code FaceForge GPU Gradio

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -20,10 +20,10 @@ realesrgan==0.3.0
20
  basicsr==1.4.2
21
 
22
  # PyTorch (GPU) - Critical for CUDA
23
- torch
24
- torchvision==0.13.1
25
  #--extra-index-url https://download.pytorch.org/whl/cu118
26
- #--extra-index-url https://download.pytorch.org/whl/cu121
27
 
28
 
29
 
 
20
  basicsr==1.4.2
21
 
22
  # PyTorch (GPU) - Critical for CUDA
23
+ torch==2.1.2
24
+ torchvision==0.16.2
25
  #--extra-index-url https://download.pytorch.org/whl/cu118
26
+ --extra-index-url https://download.pytorch.org/whl/cu121
27
 
28
 
29