Hugging Face's logo

Spaces:
fffiloni
/
diffBIR
Paused

App Files Community
4
fffiloni commited on
Commit
b0ef249
·
1 Parent(s): 2f03b3d

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -27,7 +27,7 @@ ENV HOME=/home/user \
27
  WORKDIR $HOME/app
28
 
29
  # Clone your repository or add your code to the container
30
- RUN git clone -b https://github.com/fffiloni/DiffBIR $HOME/app
31
 
32
  # Install specific versions of PyTorch, TorchText, and PyTorch Lightning
33
  RUN pip install torch==1.12.1 torchvision==0.13.1 pytorch-lightning
 
27
  WORKDIR $HOME/app
28
 
29
  # Clone your repository or add your code to the container
30
+ RUN git clone -b main https://github.com/fffiloni/DiffBIR $HOME/app
31
 
32
  # Install specific versions of PyTorch, TorchText, and PyTorch Lightning
33
  RUN pip install torch==1.12.1 torchvision==0.13.1 pytorch-lightning