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)