Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -26,7 +26,7 @@ WORKDIR $HOME/app
|
|
| 26 |
|
| 27 |
# Clone your repository or add your code to the container
|
| 28 |
RUN git clone -b dev https://github.com/camenduru/PASD $HOME/app
|
| 29 |
-
|
| 30 |
RUN pip install -q gradio gradio_imageslider diffusers==0.21.4 accelerate basicsr ultralytics salesforce-lavis webdataset pytorch_lightning
|
| 31 |
RUN pip install -q https://download.pytorch.org/whl/cu121/xformers-0.0.22.post7-cp310-cp310-manylinux2014_x86_64.whl
|
| 32 |
|
|
|
|
| 26 |
|
| 27 |
# Clone your repository or add your code to the container
|
| 28 |
RUN git clone -b dev https://github.com/camenduru/PASD $HOME/app
|
| 29 |
+
RUN pip install torchvision==0.15.2 torchaudio==2.0.2
|
| 30 |
RUN pip install -q gradio gradio_imageslider diffusers==0.21.4 accelerate basicsr ultralytics salesforce-lavis webdataset pytorch_lightning
|
| 31 |
RUN pip install -q https://download.pytorch.org/whl/cu121/xformers-0.0.22.post7-cp310-cp310-manylinux2014_x86_64.whl
|
| 32 |
|