Spaces:
Sleeping
Sleeping
Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -44,7 +44,7 @@ WORKDIR $HOME/app
|
|
| 44 |
RUN git clone -b main https://github.com/fffiloni/video-retalking $HOME/app
|
| 45 |
|
| 46 |
# Install specific versions of PyTorch and TorchVision
|
| 47 |
-
RUN pip install torchvision==0.10.0
|
| 48 |
|
| 49 |
# Install dependencies
|
| 50 |
#COPY requirements.txt $HOME/app/requirements.txt
|
|
|
|
| 44 |
RUN git clone -b main https://github.com/fffiloni/video-retalking $HOME/app
|
| 45 |
|
| 46 |
# Install specific versions of PyTorch and TorchVision
|
| 47 |
+
RUN pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html
|
| 48 |
|
| 49 |
# Install dependencies
|
| 50 |
#COPY requirements.txt $HOME/app/requirements.txt
|