Spaces:
Sleeping
Sleeping
Update Dockerfile
Browse files- Dockerfile +2 -0
Dockerfile
CHANGED
|
@@ -55,8 +55,10 @@ RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co
|
|
| 55 |
|
| 56 |
WORKDIR $HOME/app
|
| 57 |
RUN mkdir output
|
|
|
|
| 58 |
COPY app.py .
|
| 59 |
|
|
|
|
| 60 |
# Set the environment variable to specify the GPU device
|
| 61 |
ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
|
| 62 |
ENV CUDA_VISIBLE_DEVICES=0
|
|
|
|
| 55 |
|
| 56 |
WORKDIR $HOME/app
|
| 57 |
RUN mkdir output
|
| 58 |
+
COPY examples/ .
|
| 59 |
COPY app.py .
|
| 60 |
|
| 61 |
+
|
| 62 |
# Set the environment variable to specify the GPU device
|
| 63 |
ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
|
| 64 |
ENV CUDA_VISIBLE_DEVICES=0
|