initial commit

Author: akitaonrails

.gitignore

@@ -0,0 +1,2 @@
+hf_download
+outputs

Dockerfile

@@ -0,0 +1,64 @@
+# Base image with CUDA 12.6 (compatible with Torch 2.6's CUDA 12.4 requirements)
+FROM nvidia/cuda:12.6.0-devel-ubuntu22.04
+
+# Set environment variables
+ENV DEBIAN_FRONTEND=noninteractive \
+    PYTHONUNBUFFERED=1 \
+    PYTHONDONTWRITEBYTECODE=1 \
+    VIRTUAL_ENV=/venv \
+    PATH="/venv/bin:$PATH" \
+    LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH"
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    git \
+    python3.10 \
+    python3.10-venv \
+    python3.10-dev \
+    libgl1 \
+    libglib2.0-0 \
+    libsm6 \
+    libxrender1 \
+    libxext6 \
+    ninja-build \
+    && rm -rf /var/lib/apt/lists/*
+
+# Create virtual environment
+RUN python3.10 -m venv $VIRTUAL_ENV
+
+# Install PyTorch 2.6.0 with CUDA 12.4 compatibility (works with 12.6 base)
+RUN pip install --no-cache-dir \
+    torch==2.6.0 \
+    torchvision \
+    torchaudio \
+    --index-url https://download.pytorch.org/whl/cu124
+
+# Clone FramePack (Torch 2.6 compatible version)
+RUN git clone https://github.com/lllyasviel/FramePack /app && \
+    cd /app
+
+WORKDIR /app
+
+# Install requirements
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Install additional dependencies
+RUN pip install --no-cache-dir \
+    triton==3.0.0 \
+    sageattention==1.0.6
+
+# Model directory setup
+RUN mkdir -p /app/hf_download && \
+    chmod -R 777 /app/hf_download
+
+VOLUME /app/hf_download
+
+# Expose output directory
+RUN mkdir -p /app/outputs && \
+    chmod -R 777 /app/outputs
+
+VOLUME /app/outputs
+
+# Runtime configuration
+EXPOSE 7860
+CMD ["python", "demo_gradio.py", "--share"]

README.md

@@ -0,0 +1,24 @@
+## FramePack Docker CUDA
+
+Very easy:
+
+```
+
+git clone https://github.com/akitaonrails/FramePack-Docker-CUDA.git
+cd FramePack-Docker-CUDA
+mkdir outputs
+mkdir hf_download
+
+# Build the image
+docker build -t framepack-torch26-cu124:latest .
+
+# Run mapping the directories outside:
+docker run -it --rm --gpus all -p 7860:7860 \
+  -v ./outputs:/app/outputs \
+  -v ./hf_download:/app/hf_download \
+  framepack-torch26-cu124:latest
+```
+
+The first time it runs, it will download all necessary HunyuanVideo, Flux and other neccessary models. It will be more than 30GB, so be patient, but they will be cached on the external mapped directory.
+
+When it finishes access http://localhost:7860 and that's it!

entrypoint.sh

@@ -0,0 +1,18 @@
+#!/bin/bash
+set -e
+
+# Check for required models
+MODEL_CHECKLIST=(
+  "/app/models/checkpoints/sd_xl_base_1.0.safetensors"
+  "/app/models/upscale_models/realesr-general-x4v3.pth"
+)
+
+for model in "${MODEL_CHECKLIST[@]}"; do
+  if [ ! -f "$model" ]; then
+    echo "ERROR: Missing required model: $model"
+    echo "Either mount volumes or rebuild image with model downloads"
+    exit 1
+  fi
+done
+
+exec python demo_gradio.py --share --server-name 0.0.0.0