From 9626795365c26834c67c40add4f809014eef5fca Mon Sep 17 00:00:00 2001 From: Haoming Date: Mon, 8 Jun 2026 00:03:38 +0800 Subject: [PATCH] loader --- backend/loader.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/backend/loader.py b/backend/loader.py index 701219d8..5c11d5b0 100644 --- a/backend/loader.py +++ b/backend/loader.py @@ -20,6 +20,7 @@ from backend.diffusion_engine.flux import Flux from backend.diffusion_engine.flux2 import Flux2 from backend.diffusion_engine.lumina import Lumina2 from backend.diffusion_engine.mugen import Mugen +from backend.diffusion_engine.pid import PiD from backend.diffusion_engine.qwen import QwenImage from backend.diffusion_engine.sd15 import StableDiffusion from backend.diffusion_engine.sdxl import StableDiffusionXL, StableDiffusionXLRefiner @@ -41,7 +42,7 @@ from backend.utils import ( ) from modules_forge.packages.comfy.utils import convert_diffusers_mmdit -possible_models: tuple["ForgeDiffusionEngine"] = (StableDiffusion, StableDiffusionXLRefiner, StableDiffusionXL, Mugen, Chroma, Flux, Flux2, Wan, QwenImage, Lumina2, ZImage, Anima, ErnieImage) +possible_models: tuple["ForgeDiffusionEngine"] = (StableDiffusion, StableDiffusionXLRefiner, StableDiffusionXL, Mugen, Chroma, Flux, Flux2, Wan, QwenImage, Lumina2, ZImage, Anima, ErnieImage, PiD) logger = logging.getLogger("loader") setup_logger(logger)