camenduru commited on
Commit
e317038
1 Parent(s): 20b1f60

Create Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +24 -0
Dockerfile ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM runpod/pytorch:2.2.1-py3.10-cuda12.1.1-devel-ubuntu22.04
2
+ WORKDIR /content
3
+ ENV PATH="/home/camenduru/.local/bin:${PATH}"
4
+ RUN adduser --disabled-password --gecos '' camenduru && \
5
+ adduser camenduru sudo && \
6
+ echo '%sudo ALL=(ALL) NOPASSWD:ALL' >> /etc/sudoers && \
7
+ chown -R camenduru:camenduru /content && \
8
+ chmod -R 777 /content && \
9
+ chown -R camenduru:camenduru /home && \
10
+ chmod -R 777 /home
11
+
12
+ RUN 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 camenduru
15
+
16
+ RUN pip install -q opencv-python imageio imageio-ffmpeg ffmpeg-python av runpod \
17
+ xformers==0.0.25 git+https://github.com/camenduru/Lumina-T2X@dev && \
18
+ git clone -b dev https://github.com/camenduru/Lumina-T2X /content/Lumina-T2X && \
19
+ aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/Lumina-Next-SFT/resolve/main/consolidated_ema.00-of-01.safetensors -d /content/Lumina-T2X/models -o consolidated_ema.00-of-01.safetensors && \
20
+ aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/Lumina-Next-SFT/resolve/main/model_args.pth -d /content/Lumina-T2X/models -o model_args.pth
21
+
22
+ COPY ./worker_runpod.py /content/Lumina-T2X/lumina_next_compositional_generation/worker_runpod.py
23
+ WORKDIR /content/Lumina-T2X/lumina_next_compositional_generation
24
+ CMD python worker_runpod.py