diff --git a/Dockerfile b/Dockerfile index a2f2731..ef343a2 100644 --- a/Dockerfile +++ b/Dockerfile @@ -5,10 +5,11 @@ FROM runpod/base:0.6.1-cuda12.2.0 RUN apt-get update && apt-get install -y ffmpeg aria2 git unzip # Install Python dependencies -RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 \ - opencv-python imageio imageio-ffmpeg ffmpeg-python av runpod \ +RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 && \ + pip install --no-cache-dir opencv-python imageio imageio-ffmpeg ffmpeg-python av runpod \ xformers==0.0.25 torchsde==0.2.6 einops==0.8.0 diffusers==0.28.0 transformers==4.41.2 accelerate==0.30.1 + # Clone the ComfyUI repository RUN git clone https://github.com/comfyanonymous/ComfyUI /content/ComfyUI diff --git a/flux1-dev-lora-worker-project/.runpodignore b/flux1-dev-lora-worker-project/.runpodignore new file mode 100644 index 0000000..e4a6598 --- /dev/null +++ b/flux1-dev-lora-worker-project/.runpodignore @@ -0,0 +1,10 @@ +# Similar to .gitignore +# Matches do not sync to the Project Pod or cause the Pod to reload. + +Dockerfile +__pycache__/ +*.pyc +.*.swp +.git/ +*.tmp +*.log \ No newline at end of file