Update Dockerfile
Browse files- Dockerfile +3 -18
Dockerfile
CHANGED
@@ -1,9 +1,6 @@
|
|
1 |
FROM runpod/pytorch:2.2.1-py3.10-cuda12.1.1-devel-ubuntu22.04
|
2 |
-
#FROM mybigpai-public-registry.cn-beijing.cr.aliyuncs.com/easycv/torch_cuda:cogvideox_fun
|
3 |
-
#FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
|
4 |
WORKDIR /content
|
5 |
ENV PATH="/home/zebraslive/.local/bin:${PATH}"
|
6 |
-
ENV DEBIAN_FRONTEND noninteractive
|
7 |
|
8 |
RUN adduser --disabled-password --gecos '' zebraslive && \
|
9 |
adduser zebraslive sudo && \
|
@@ -14,13 +11,9 @@ RUN adduser --disabled-password --gecos '' zebraslive && \
|
|
14 |
chmod -R 777 /home && \
|
15 |
apt update -y && add-apt-repository -y ppa:git-core/ppa && apt update -y && apt install -y aria2 git git-lfs unzip ffmpeg
|
16 |
|
17 |
-
RUN apt-get install -y nginx
|
18 |
-
EXPOSE 80
|
19 |
-
|
20 |
USER zebraslive
|
21 |
-
|
22 |
-
COPY
|
23 |
-
RUN pip install -r /content/requirements.txt
|
24 |
|
25 |
RUN pip install -q torch==2.4.0+cu121 torchvision==0.19.0+cu121 torchaudio==2.4.0+cu121 torchtext==0.18.0 torchdata==0.8.0 --extra-index-url https://download.pytorch.org/whl/cu121 \
|
26 |
tqdm==4.66.5 numpy==1.26.3 imageio==2.35.1 imageio-ffmpeg==0.5.1 xformers==0.0.27.post2 diffusers==0.30.3 moviepy==1.0.3 transformers==4.44.2 accelerate==0.33.0 sentencepiece==0.2.0 pillow==9.5.0 runpod && \
|
@@ -42,13 +35,5 @@ RUN pip install -q torch==2.4.0+cu121 torchvision==0.19.0+cu121 torchaudio==2.4.
|
|
42 |
aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/alibaba-pai/CogVideoX-Fun-V1.1-5b-InP/raw/main/model_index.json -d /content/model -o model_index.json
|
43 |
|
44 |
COPY ./worker_runpod.py /content/worker_runpod.py
|
45 |
-
COPY ./cogvideox /content/cogvideox
|
46 |
-
COPY ./asset /content/asset
|
47 |
-
COPY ./config /content/config
|
48 |
-
COPY ./datasets /content/datasets
|
49 |
-
COPY ./reports /content/reports
|
50 |
-
|
51 |
-
# Copy the README.md
|
52 |
-
COPY README.md /usr/share/nginx/html/README.md
|
53 |
WORKDIR /content
|
54 |
-
CMD
|
|
|
1 |
FROM runpod/pytorch:2.2.1-py3.10-cuda12.1.1-devel-ubuntu22.04
|
|
|
|
|
2 |
WORKDIR /content
|
3 |
ENV PATH="/home/zebraslive/.local/bin:${PATH}"
|
|
|
4 |
|
5 |
RUN adduser --disabled-password --gecos '' zebraslive && \
|
6 |
adduser zebraslive sudo && \
|
|
|
11 |
chmod -R 777 /home && \
|
12 |
apt update -y && add-apt-repository -y ppa:git-core/ppa && apt update -y && apt install -y aria2 git git-lfs unzip ffmpeg
|
13 |
|
|
|
|
|
|
|
14 |
USER zebraslive
|
15 |
+
RUN git clone https://github.com/aigc-apps/CogVideoX-Fun
|
16 |
+
COPY CogVideoX-Fun /content
|
|
|
17 |
|
18 |
RUN pip install -q torch==2.4.0+cu121 torchvision==0.19.0+cu121 torchaudio==2.4.0+cu121 torchtext==0.18.0 torchdata==0.8.0 --extra-index-url https://download.pytorch.org/whl/cu121 \
|
19 |
tqdm==4.66.5 numpy==1.26.3 imageio==2.35.1 imageio-ffmpeg==0.5.1 xformers==0.0.27.post2 diffusers==0.30.3 moviepy==1.0.3 transformers==4.44.2 accelerate==0.33.0 sentencepiece==0.2.0 pillow==9.5.0 runpod && \
|
|
|
35 |
aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/alibaba-pai/CogVideoX-Fun-V1.1-5b-InP/raw/main/model_index.json -d /content/model -o model_index.json
|
36 |
|
37 |
COPY ./worker_runpod.py /content/worker_runpod.py
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
38 |
WORKDIR /content
|
39 |
+
CMD python worker_runpod.py
|