Merge pull request #15 from XBastille/testing_2

added NVIDIA CUDA Docker support

Author: XBastille

Dockerfile.gpu

@@ -0,0 +1,92 @@
+FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
+
+ENV PYTHONDONTWRITEBYTECODE 1
+ENV PYTHONUNBUFFERED 1
+ENV NVIDIA_VISIBLE_DEVICES all
+ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
+
+WORKDIR /app
+
+ARG NODE_MAJOR=22
+
+RUN apt-get update \
+    && apt-get install -y --no-install-recommends \
+        ca-certificates \
+        curl \
+        gnupg \
+        build-essential \
+        libglib2.0-0 \
+        libsm6 \
+        libxext6 \
+        libxrender1 \
+        libgomp1 \
+        libfontconfig1 \
+        ffmpeg \
+        libgl1-mesa-glx \
+        libgl1-mesa-dri \
+        libglu1-mesa \
+        python3 \
+        python3-pip \
+        python3-dev \
+    && mkdir -p /etc/apt/keyrings \
+    && curl -fsSL https://deb.nodesource.com/gpgkey/nodesource-repo.gpg.key | gpg --dearmor -o /etc/apt/keyrings/nodesource.gpg \
+    && echo "deb [signed-by=/etc/apt/keyrings/nodesource.gpg] https://deb.nodesource.com/node_$NODE_MAJOR.x nodistro main" | tee /etc/apt/sources.list.d/nodesource.list \
+    && apt-get update \
+    && apt-get install -y --no-install-recommends nodejs \
+    && rm -rf /var/lib/apt/lists/* /usr/share/doc /usr/share/man \
+    && apt-get clean \
+    && useradd --create-home python \
+    && chown python:python -R /app
+
+USER python
+
+COPY --chown=python:python requirements*.txt ./
+
+RUN pip3 install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
+
+RUN pip3 install --no-cache-dir -r requirements.txt
+
+ENV DEBUG="false" \
+    PYTHONUNBUFFERED="true" \
+    PYTHONDONTWRITEBYTECODE="1" \
+    PATH="${PATH}:/home/python/.local/bin" \
+    USER="python" \
+    QT_QPA_PLATFORM="offscreen" \
+    CUDA_VISIBLE_DEVICES="0" \
+    HF_HOME="/app/.cache/huggingface" \
+    WEB_CONCURRENCY="2"
+
+COPY --chown=python:python package.json tailwind.config.js ./
+
+RUN npm install
+
+COPY --chown=python:python . .
+
+RUN mkdir -p static/css staticfiles media tmp
+
+RUN chmod +x ./startup.sh
+
+RUN chmod +x ./download-models.sh
+RUN ./download-models.sh
+
+ARG HF_TOKEN
+ENV HF_TOKEN=${HF_TOKEN}
+RUN if [ -n "$HF_TOKEN" ]; then \
+        echo "Downloading HuggingFace models..."; \
+        python3 download-hf-models.py; \
+    else \
+        echo "HF_TOKEN not provided - models will be downloaded at runtime"; \
+    fi
+
+RUN npm run build
+
+RUN python3 manage.py collectstatic --no-input --verbosity 2
+
+RUN python3 manage.py migrate
+
+HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
+  CMD curl -f http://localhost:8000/admin/login/ || exit 1
+
+EXPOSE 8000
+
+CMD ["./startup.sh"]

ai_image_editor/models/apply_fill.py

@@ -83,21 +83,3 @@ def apply_fill(
 
     result = result.resize(original_size)
     result.save(img_filled_p)
-
-
-if __name__ == "__main__":
-    input_img = "test.jpg"
-    img_stem = Path(input_img).stem
-    masks_dir = f"./results/{img_stem}"
-    text_prompt = "front view of a blond woman with a warm smile"
-
-    apply_fill(
-        input_img=input_img,
-        masks_dir=masks_dir,
-        text_prompt=text_prompt,
-        controlnet_scale=0.9,
-        guidance_scale=3.5,
-        num_inference_steps=40,
-        negative_prompt="Low quality, Blurry, Overexposed, Underexposed, Grainy, Pixelated, Low resolution, Distorted, Over-saturated, Washed out",
-        seed=None,
-    )

ai_image_editor/models/transformer_flux.py

@@ -520,4 +520,4 @@ def custom_forward(*inputs):
         if not return_dict:
             return (output,)
 
-        return Transformer2DModelOutput(sample=output)
+        return Transformer2DModelOutput(sample=output)

ai_image_editor/views.py

@@ -148,4 +148,4 @@ def process_image(request):
 
         return JsonResponse(
             {"status": "success", "result": f"data:image/png;base64,{img_str}"}
-        )
+        )

build-gpu.bat

@@ -0,0 +1,22 @@
+@echo off
+
+echo  Building DeepFX Studio GPU Docker image...
+
+if "%HF_TOKEN%"=="" (
+    echo    Warning: HF_TOKEN environment variable not set!
+    echo    Gated models (SD 3.5, FLUX) will be downloaded at runtime.
+    echo    To pre-download models, set HF_TOKEN first:
+    echo    set HF_TOKEN=your_huggingface_token
+    echo.
+    
+    docker build -f Dockerfile.gpu -t deepfx-studio-gpu .
+) else (
+    echo   HF_TOKEN found - will pre-download gated models
+    
+    docker build -f Dockerfile.gpu --build-arg HF_TOKEN=%HF_TOKEN% -t deepfx-studio-gpu .
+)
+
+echo.
+echoBuild completed!
+echoTo run: docker run --gpus all -p 8000:8000 deepfx-studio-gpu
+pause

build-gpu.sh

@@ -0,0 +1,25 @@
+#!/bin/bash
+
+set -e
+
+echo "Building DeepFX Studio GPU Docker image..."
+
+if [ -z "$HF_TOKEN" ]; then
+    echo "Warning: HF_TOKEN environment variable not set!"
+    echo "   Gated models (SD 3.5, FLUX) will be downloaded at runtime."
+    echo "   To pre-download models, set HF_TOKEN first:"
+    echo "   export HF_TOKEN=your_huggingface_token"
+    echo ""
+    
+    docker build -f Dockerfile.gpu -t deepfx-studio-gpu .
+else
+    echo "HF_TOKEN found - will pre-download gated models"
+    
+    docker build -f Dockerfile.gpu \
+        --build-arg HF_TOKEN="$HF_TOKEN" \
+        -t deepfx-studio-gpu .
+fi
+
+echo ""
+echo "Build completed!"
+echo "To run: docker run --gpus all -p 8000:8000 deepfx-studio-gpu"

core/__init__.py

@@ -0,0 +1,3 @@
+"""
+Core Django app for DeepFX Studio dependency management
+"""

core/admin.py

@@ -0,0 +1,3 @@
+from django.contrib import admin
+
+# Register your models here.

core/apps.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class CoreConfig(AppConfig):
+    default_auto_field = 'django.db.models.BigAutoField'
+    name = 'core'