fix huggingface loading

This commit is contained in:
Zeqiang Lai
2025-06-14 10:35:18 +08:00
committed by GitHub
parent 52e68bf758
commit dfcd5c829d
4 changed files with 3 additions and 4 deletions

View File

@@ -45,8 +45,8 @@ from diffusers.utils import deprecate
from diffusers.callbacks import MultiPipelineCallbacks, PipelineCallback from diffusers.callbacks import MultiPipelineCallbacks, PipelineCallback
from diffusers.image_processor import PipelineImageInput from diffusers.image_processor import PipelineImageInput
from diffusers.pipelines.stable_diffusion.pipeline_output import StableDiffusionPipelineOutput from diffusers.pipelines.stable_diffusion.pipeline_output import StableDiffusionPipelineOutput
from .modules import UNet2p5DConditionModel from .unet.modules import UNet2p5DConditionModel
from .attn_processor import SelfAttnProcessor2_0, RefAttnProcessor2_0, PoseRoPEAttnProcessor2_0 from .unet.attn_processor import SelfAttnProcessor2_0, RefAttnProcessor2_0, PoseRoPEAttnProcessor2_0
__all__ = [ __all__ = [
"HunyuanPaintPipeline", "HunyuanPaintPipeline",

View File

@@ -32,9 +32,8 @@ from diffusers import (
UNet2DConditionModel, UNet2DConditionModel,
ControlNetModel, ControlNetModel,
) )
from .pipeline import UNet2p5DConditionModel
from .modules import Dino_v2 from .modules import Dino_v2, UNet2p5DConditionModel
import math import math