deepbeepmeep · Tophness · Jun 20, 2025 · Jul 9, 2025 · Jul 11, 2025 · Aug 30, 2025
diff --git a/.gitignore b/.gitignore
@@ -41,3 +41,7 @@ gradio_outputs/
 ckpts/
 loras/
 loras_i2v/
+
+settings/
+
+wgp_config.json
diff --git a/Dockerfile b/Dockerfile
@@ -0,0 +1,92 @@
+FROM nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04
+
+# Build arg for GPU architectures - specify which CUDA compute capabilities to compile for
+# Common values:
+#   7.0  - Tesla V100
+#   7.5  - RTX 2060, 2070, 2080, Titan RTX
+#   8.0  - A100, A800 (Ampere data center)
+#   8.6  - RTX 3060, 3070, 3080, 3090 (Ampere consumer)
+#   8.9  - RTX 4070, 4080, 4090 (Ada Lovelace)
+#   9.0  - H100, H800 (Hopper data center)
+#   12.0 - RTX 5070, 5080, 5090 (Blackwell) - Note: sm_120 architecture
+#
+# Examples:
+#   RTX 3060: --build-arg CUDA_ARCHITECTURES="8.6"
+#   RTX 4090: --build-arg CUDA_ARCHITECTURES="8.9"
+#   Multiple: --build-arg CUDA_ARCHITECTURES="8.0;8.6;8.9"
+#
+# Note: Including 8.9 or 9.0 may cause compilation issues on some setups
+# Default includes 8.0 and 8.6 for broad Ampere compatibility
+ARG CUDA_ARCHITECTURES="8.0;8.6"
+
+ENV DEBIAN_FRONTEND=noninteractive
+
+# Install system dependencies
+RUN apt update && \
+    apt install -y \
+    python3 python3-pip git wget curl cmake ninja-build \
+    libgl1 libglib2.0-0 ffmpeg && \
+    apt clean
+
+WORKDIR /workspace
+
+COPY requirements.txt .
+
+# Upgrade pip first
+RUN pip install --upgrade pip setuptools wheel
+
+# Install requirements if exists
+RUN pip install -r requirements.txt
+
+# Install PyTorch with CUDA support
+RUN pip install --extra-index-url https://download.pytorch.org/whl/cu124 \
+    torch==2.6.0+cu124 torchvision==0.21.0+cu124
+
+# Install SageAttention from git (patch GPU detection)
+ENV TORCH_CUDA_ARCH_LIST="${CUDA_ARCHITECTURES}"
+ENV FORCE_CUDA="1"
+ENV MAX_JOBS="1"
+
+COPY <<EOF /tmp/patch_setup.py
+import os
+with open('setup.py', 'r') as f:
+    content = f.read()
+
+# Get architectures from environment variable
+arch_list = os.environ.get('TORCH_CUDA_ARCH_LIST')
+arch_set = '{' + ', '.join([f'"{arch}"' for arch in arch_list.split(';')]) + '}'
+
+# Replace the GPU detection section
+old_section = '''compute_capabilities = set()
+device_count = torch.cuda.device_count()
+for i in range(device_count):
+    major, minor = torch.cuda.get_device_capability(i)
+    if major < 8:
+        warnings.warn(f"skipping GPU {i} with compute capability {major}.{minor}")
+        continue
+    compute_capabilities.add(f"{major}.{minor}")'''
+
+new_section = 'compute_capabilities = ' + arch_set + '''
+print(f"Manually set compute capabilities: {compute_capabilities}")'''
+
+content = content.replace(old_section, new_section)
+
+with open('setup.py', 'w') as f:
+    f.write(content)
+EOF
+
+RUN git clone https://github.com/thu-ml/SageAttention.git /tmp/sageattention && \
+    cd /tmp/sageattention && \
+    python3 /tmp/patch_setup.py && \
+    pip install --no-build-isolation .
+
+RUN useradd -u 1000 -ms /bin/bash user
+
+RUN chown -R user:user /workspace
+
+RUN mkdir /home/user/.cache && \
+    chown -R user:user /home/user/.cache
+
+COPY entrypoint.sh /workspace/entrypoint.sh
+
+ENTRYPOINT ["/workspace/entrypoint.sh"]
diff --git a/README.md b/README.md
diff --git a/defaults/animate.json b/defaults/animate.json
@@ -0,0 +1,17 @@
+{
+    "model": {
+        "name": "Wan2.2 Animate 14B",
+        "architecture": "animate",
+        "description": "Wan-Animate takes a video and a character image as input, and generates a video in either 'Animation' or 'Replacement' mode. Sliding Window of 81 frames at least are recommeded to obtain the best Style continuity.",
+        "URLs": [
+            "https://huggingface.co/DeepBeepMeep/Wan2.2/resolve/main/wan2.2_animate_14B_bf16.safetensors",
+            "https://huggingface.co/DeepBeepMeep/Wan2.2/resolve/main/wan2.2_animate_14B_quanto_fp16_int8.safetensors",
+            "https://huggingface.co/DeepBeepMeep/Wan2.2/resolve/main/wan2.2_animate_14B_quanto_bf16_int8.safetensors"
+        ],
+		"preload_URLs" :
+		[
+			"https://huggingface.co/DeepBeepMeep/Wan2.2/resolve/main/wan2.2_animate_relighting_lora.safetensors"
+		],
+        "group": "wan2_2"
+    }
+}
diff --git a/defaults/fantasy.json b/defaults/fantasy.json
@@ -1,7 +1,7 @@
 {
 	"model":
 	{
-		"name": "Fantasy Talking 720p",
+		"name": "Fantasy Talking 720p 14B",
 		"architecture" : "fantasy",
 		"modules": [ ["https://huggingface.co/DeepBeepMeep/Wan2.1/resolve/main/wan2.1_fantasy_speaking_14B_bf16.safetensors"]],
 		"description": "The Fantasy Talking model corresponds to the original Wan image 2 video model combined with the Fantasy Speaking module to process an audio Input.",

diff --git a/defaults/flux.json b/defaults/flux.json
@@ -7,8 +7,7 @@
             "https://huggingface.co/DeepBeepMeep/Flux/resolve/main/flux1-dev_bf16.safetensors",
             "https://huggingface.co/DeepBeepMeep/Flux/resolve/main/flux1-dev_quanto_bf16_int8.safetensors"
         ],
-        "image_outputs": true,
-        "flux-model": "flux-dev"
+        "image_outputs": true
     },
     "prompt": "draw a hat",
     "resolution": "1280x720",

diff --git a/defaults/flux_chroma.json b/defaults/flux_chroma.json
@@ -1,14 +1,13 @@
 {
     "model": {
         "name": "Flux 1 Chroma 1 HD 8.9B",
-        "architecture": "flux",
+        "architecture": "flux_chroma",
         "description": "FLUX.1 Chroma is a 8.9 billion parameters model. As a base model, Chroma1 is intentionally designed to be an excellent starting point for finetuning. It provides a strong, neutral foundation for developers, researchers, and artists to create specialized models..",
         "URLs": [
             "https://huggingface.co/DeepBeepMeep/Flux/resolve/main/flux1-chroma_hd_bf16.safetensors",
             "https://huggingface.co/DeepBeepMeep/Flux/resolve/main/flux1-chroma_hd_quanto_bf16_int8.safetensors"
         ],
-        "image_outputs": true,
-        "flux-model": "flux-chroma"
+        "image_outputs": true
     },
     "prompt": "draw a hat",
     "resolution": "1280x720",

diff --git a/defaults/flux_dev_kontext.json b/defaults/flux_dev_kontext.json
@@ -1,15 +1,12 @@
 {
     "model": {
         "name": "Flux 1 Dev Kontext 12B",
-        "architecture": "flux",
+        "architecture": "flux_dev_kontext",
         "description": "FLUX.1 Kontext is a 12 billion parameter rectified flow transformer capable of editing images based on instructions stored in the Prompt. Please be aware that Flux Kontext is picky on the resolution of the input image and the output dimensions may not match the dimensions of the input image.",
         "URLs": [
             "https://huggingface.co/DeepBeepMeep/Flux/resolve/main/flux1_kontext_dev_bf16.safetensors",
             "https://huggingface.co/DeepBeepMeep/Flux/resolve/main/flux1_kontext_dev_quanto_bf16_int8.safetensors"
-        ],
-		"image_outputs": true,		
-		"reference_image": true,		
-		"flux-model": "flux-dev-kontext"		
+        ]
     },
 	"prompt": "add a hat",
     "resolution": "1280x720",

diff --git a/defaults/flux_dev_umo.json b/defaults/flux_dev_umo.json
@@ -0,0 +1,23 @@
+{
+    "model": {
+        "name": "Flux 1 UMO Dev 12B",
+        "architecture": "flux_dev_umo",
+        "description": "FLUX.1 UMO Dev is a model that can Edit Images with a specialization in combining multiple image references (resized internally at 512x512 max) to produce an Image output. Best Image preservation at 768x768 Resolution Output.",
+        "URLs": "flux",
+		"loras": ["https://huggingface.co/DeepBeepMeep/Flux/resolve/main/flux1-dev-UMO_dit_lora_bf16.safetensors"],
+		"resolutions":  [ ["1024x1024 (1:1)", "1024x1024"],
+						["768x1024 (3:4)", "768x1024"],
+						["1024x768 (4:3)", "1024x768"],
+						["512x1024 (1:2)", "512x1024"],
+						["1024x512 (2:1)", "1024x512"],
+						["768x768 (1:1)", "768x768"],
+						["768x512 (3:2)", "768x512"],
+						["512x768 (2:3)", "512x768"]]
+    },	
+	"prompt": "the man is wearing a hat",
+	"embedded_guidance_scale": 4,
+    "resolution": "768x768",
+    "batch_size": 1
+}
+
+
diff --git a/defaults/flux_dev_uso.json b/defaults/flux_dev_uso.json
@@ -1,16 +1,13 @@
 {
     "model": {
-        "name": "Flux 1 Dev USO 12B",
-        "architecture": "flux",
-        "description": "FLUX.1 Dev USO is a model specialized to Edit Images with a specialization in Style Transfers (up to two).",
+        "name": "Flux 1 USO Dev 12B",
+        "architecture": "flux_dev_uso",
+        "description": "FLUX.1 USO Dev is a model that can Edit Images with a specialization in Style Transfers (up to two).",
 		"modules": [ ["https://huggingface.co/DeepBeepMeep/Flux/resolve/main/flux1-dev-USO_projector_bf16.safetensors"]],
         "URLs": "flux",
-		"loras": ["https://huggingface.co/DeepBeepMeep/Flux/resolve/main/flux1-dev-USO_dit_lora_bf16.safetensors"],
-		"image_outputs": true,		
-		"reference_image": true,		
-		"flux-model": "flux-dev-uso"		
+		"loras": ["https://huggingface.co/DeepBeepMeep/Flux/resolve/main/flux1-dev-USO_dit_lora_bf16.safetensors"]
     },
-	"prompt": "add a hat",
+	"prompt": "the man is wearing a hat",
 	"embedded_guidance_scale": 4,
     "resolution": "1024x1024",
     "batch_size": 1

diff --git a/defaults/flux_krea.json b/defaults/flux_krea.json
@@ -1,14 +1,13 @@
 {
     "model": {
-        "name": "Flux 1 Krea Dev 12B",
+        "name": "Flux 1 Dev Krea 12B",
         "architecture": "flux",
         "description": "Cutting-edge output quality, with a focus on aesthetic photography..",
         "URLs": [
             "https://huggingface.co/DeepBeepMeep/Flux/resolve/main/flux1-krea-dev_bf16.safetensors",
             "https://huggingface.co/DeepBeepMeep/Flux/resolve/main/flux1-krea-dev_quanto_bf16_int8.safetensors"
         ],
-        "image_outputs": true,
-        "flux-model": "flux-dev"
+        "image_outputs": true
     },
     "prompt": "draw a hat",
     "resolution": "1280x720",

diff --git a/defaults/flux_schnell.json b/defaults/flux_schnell.json
@@ -1,14 +1,13 @@
 {
     "model": {
         "name": "Flux 1 Schnell 12B",
-        "architecture": "flux",
+        "architecture": "flux_schnell",
         "description": "FLUX.1 Schnell is a 12 billion parameter rectified flow transformer capable of generating images from text descriptions. As a distilled model it requires fewer denoising steps.",
         "URLs": [
             "https://huggingface.co/DeepBeepMeep/Flux/resolve/main/flux1-schnell_bf16.safetensors",
             "https://huggingface.co/DeepBeepMeep/Flux/resolve/main/flux1-schnell_quanto_bf16_int8.safetensors"
         ],
-        "image_outputs": true,
-        "flux-model": "flux-schnell"
+        "image_outputs": true
     },
     "prompt": "draw a hat",
     "resolution": "1280x720",

diff --git a/defaults/flux_srpo.json b/defaults/flux_srpo.json
@@ -0,0 +1,14 @@
+{
+    "model": {
+        "name": "Flux 1 Dev SRPO 12B",
+        "architecture": "flux",
+        "description": "By fine-tuning the FLUX.1.dev model with optimized denoising and online reward adjustment, SRPO improves its human-evaluated realism and aesthetic quality by over 3x.",
+        "URLs": [
+            "https://huggingface.co/DeepBeepMeep/Flux/resolve/main/flux1-srpo-dev_bf16.safetensors",
+            "https://huggingface.co/DeepBeepMeep/Flux/resolve/main/flux1-srpo-dev_quanto_bf16_int8.safetensors"
+        ]
+    },
+    "prompt": "draw a hat",
+    "resolution": "1024x1024",
+    "batch_size": 1
+}
diff --git a/defaults/flux_srpo_uso.json b/defaults/flux_srpo_uso.json
@@ -0,0 +1,16 @@
+{
+    "model": {
+        "name": "Flux 1 USO SRPO 12B",
+        "architecture": "flux_dev_uso",
+        "description": "FLUX.1 USO SRPO is a model that can Edit Images with a specialization in Style Transfers (up to two). It leverages the improved Image quality brought by the SRPO process",
+		"modules": [ "flux_dev_uso"],
+        "URLs": "flux_srpo",
+		"loras": "flux_dev_uso"
+    },
+	"prompt": "the man is wearing a hat",
+	"embedded_guidance_scale": 4,
+    "resolution": "1024x1024",
+    "batch_size": 1
+}
+
+
diff --git a/defaults/hunyuan_t2v_accvideo.json b/defaults/hunyuan_t2v_accvideo.json
@@ -1,6 +1,6 @@
 {
   "model": {
-    "name": "Hunyuan Video AccVideo 720p 13B",
+    "name": "Hunyuan Video Text2video 720p AccVideo 13B",
     "architecture": "hunyuan",
     "description": " AccVideo is a novel efficient distillation method to accelerate video diffusion models with synthetic datset. Our method is 8.5x faster than HunyuanVideo.",
     "URLs": [

diff --git a/defaults/hunyuan_t2v_fast.json b/defaults/hunyuan_t2v_fast.json
@@ -1,8 +1,9 @@
 {
   "model": {
-    "name": "Hunyuan Video FastHunyuan 720p 13B",
+    "name": "Hunyuan Video Text2video 720p FastHunyuan 13B",
     "architecture": "hunyuan",
     "description": "Fast Hunyuan is an accelerated HunyuanVideo model. It can sample high quality videos with 6 diffusion steps.",
+	"settings_dir": [ "" ],
     "URLs": [
       "https://huggingface.co/DeepBeepMeep/HunyuanVideo/resolve/main/fast_hunyuan_video_720_quanto_int8.safetensors"
     ],

diff --git a/defaults/i2v_fusionix.json b/defaults/i2v_fusionix.json
@@ -5,6 +5,7 @@
 		"architecture" : "i2v",
 		"description": "A powerful merged image-to-video model based on the original WAN 2.1 I2V model, enhanced using multiple open-source components and LoRAs to boost motion realism, temporal consistency, and expressive detail.",
 		"URLs": "i2v",
+		"settings_dir": [ "" ],		
 		"loras": ["https://huggingface.co/DeepBeepMeep/Wan2.1/resolve/main/loras_accelerators/Wan2.1_I2V_14B_FusionX_LoRA.safetensors"]
 	}
 }
diff --git a/defaults/i2v_palingenesis_2_2.json b/defaults/i2v_palingenesis_2_2.json
@@ -0,0 +1,18 @@
+{
+	"model":
+	{
+		"name": "Wan2.2 Image2video Palingenesis 14B",
+		"architecture" : "i2v_2_2",
+		"description": "Wan 2.2 Image 2 Video model. Contrary to the Wan Image2video 2.1 this model is structurally close to the t2v model. Palingenesis a finetune praised for its high quality.",
+		"URLs": [ "https://huggingface.co/eddy1111111/WAN22.XX_Palingenesis/resolve/main/WAN22.XX_Palingenesis_high_i2v_fix.safetensors"],
+		"URLs2": [ "https://huggingface.co/eddy1111111/WAN22.XX_Palingenesis/resolve/main/WAN22.XX_Palingenesis_low_i2v_fix.safetensors"],
+		"group": "wan2_2"		
+	},
+	"ignore_unused_weights": true,
+	"guidance_phases": 2,	
+	"switch_threshold" : 900,
+	"guidance_scale" : 3.5,
+	"guidance2_scale" : 3.5,
+	"flow_shift" : 5
+
+}
diff --git a/defaults/infinitetalk.json b/defaults/infinitetalk.json
@@ -1,6 +1,6 @@
 {
     "model": {
-        "name": "Infinitetalk Single Speaker 480p",
+        "name": "Infinitetalk Single Speaker 480p 14B",
         "architecture": "infinitetalk",
         "modules": [
             [

diff --git a/defaults/infinitetalk_multi.json b/defaults/infinitetalk_multi.json
@@ -1,6 +1,6 @@
 {
     "model": {
-        "name": "Infinitetalk Multi Speakers 480p",
+        "name": "Infinitetalk Multi Speakers 480p 14B",
         "architecture": "infinitetalk",
         "modules": [
             [

diff --git a/defaults/lucy_edit.json b/defaults/lucy_edit.json
@@ -0,0 +1,20 @@
+{
+    "model": {
+        "name": "Wan2.2 Lucy Edit 5B",
+        "architecture": "lucy_edit",
+        "description": "Lucy Edit is a video editing model that performs instruction-guided edits on videos using free-text prompts. It supports a variety of edits, such as clothing & accessory changes, character changes, object insertions, and scene replacements while preserving the motion and composition perfectly.",
+        "URLs": [
+            "https://huggingface.co/DeepBeepMeep/Wan2.2/resolve/main/wan2.2_lucy_edit_mbf16.safetensors",
+            "https://huggingface.co/DeepBeepMeep/Wan2.2/resolve/main/wan2.2_lucy_edit_quanto_mbf16_int8.safetensors",
+            "https://huggingface.co/DeepBeepMeep/Wan2.2/resolve/main/wan2.2_lucy_edit_quanto_mfp16_int8.safetensors"
+        ],
+		"settings_dir": "ti2v_2_2",		
+        "group": "wan2_2"
+    },
+	"prompt": "change the clothes to red",	
+    "video_length": 81,
+    "guidance_scale": 5,
+    "flow_shift": 5,
+    "num_inference_steps": 30,
+    "resolution": "1280x720"
+}
diff --git a/defaults/lucy_edit_fastwan.json b/defaults/lucy_edit_fastwan.json
@@ -0,0 +1,17 @@
+{
+    "model": {
+        "name": "Wan2.2 Lucy Edit FastWan 5B",
+        "architecture": "lucy_edit",
+        "description": "Lucy Edit is a video editing model that performs instruction-guided edits on videos using free-text prompts. It supports a variety of edits, such as clothing & accessory changes, character changes, object insertions, and scene replacements while preserving the motion and composition perfectly. This is the FastWan version for faster generation.",
+        "URLs": "lucy_edit",
+        "group": "wan2_2",
+		"settings_dir": [ "" ],		
+		"loras": "ti2v_2_2_fastwan"
+    },
+	"prompt": "change the clothes to red",
+    "video_length": 81,
+    "guidance_scale": 1,
+    "flow_shift": 3,
+    "num_inference_steps": 5,
+    "resolution": "1280x720"
+}