AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

use unwrap_module for torch compiled modules

Browse Source
This commit is contained in:
Aryan
2025-04-14 17:34:10 +02:00
parent 2238f55f40
commit 52b9b61f62
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_modular.py
View File

@@ -30,7 +30,7 @@ from ...utils import (
scale_lora_layers,
unscale_lora_layers,
)
from ...utils.torch_utils import is_compiled_module, randn_tensor
from ...utils.torch_utils import randn_tensor, unwrap_module
from ..controlnet.multicontrolnet import MultiControlNetModel
from ..modular_pipeline import (
AutoPipelineBlocks,
@@ -2545,7 +2545,7 @@ class StableDiffusionXLControlNetDenoiseStep(PipelineBlock):
data.height = data.height * pipeline.vae_scale_factor
data.width = data.width * pipeline.vae_scale_factor
controlnet = pipeline.controlnet._orig_mod if is_compiled_module(pipeline.controlnet) else pipeline.controlnet
controlnet = unwrap_module(pipeline.controlnet)
# (1.1)
# control_guidance_start/control_guidance_end (align format)
@@ -2973,7 +2973,7 @@ class StableDiffusionXLControlNetUnionDenoiseStep(PipelineBlock):
data.height = data.height * pipeline.vae_scale_factor
data.width = data.width * pipeline.vae_scale_factor
controlnet = pipeline.controlnet._orig_mod if is_compiled_module(pipeline.controlnet) else pipeline.controlnet
controlnet = unwrap_module(pipeline.controlnet)
# (1.1)
# control guidance