meepmoo commited on
Commit
4e7576b
1 Parent(s): 1999e9d

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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 ./requirements.txt /content/requirements.txt
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 ["python3", "/content/worker_runpod.py"]
 
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