Hugging Face's logo

Spaces:
fffiloni
/
ProPainter
Paused

App Files Community
4
fffiloni commited on
Commit
8ec2392
·
verified ·
1 Parent(s): bb0efa4

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -29,7 +29,7 @@ WORKDIR $HOME/app
29
  RUN git clone https://github.com/sczhou/ProPainter.git $HOME/app
30
 
31
  # Install specific versions of PyTorch and TorchVision
32
- RUN pip install torch==2.0.1+cu117 torchvision==0.15.2+cu117 -f https://download.pytorch.org/whl/torch_stable.html
33
 
34
  # Install dependencies
35
  RUN pip install --no-cache-dir -r requirements.txt
 
29
  RUN git clone https://github.com/sczhou/ProPainter.git $HOME/app
30
 
31
  # Install specific versions of PyTorch and TorchVision
32
+ RUN pip install torch==2.5.1 torchvision==0.20.1 -f https://download.pytorch.org/whl/torch_stable.html
33
 
34
  # Install dependencies
35
  RUN pip install --no-cache-dir -r requirements.txt