fix huggingface loading
This commit is contained in:
@@ -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",
|
||||||
|
|||||||
@@ -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
|
||||||
|
|
||||||
|
|
||||||
Reference in New Issue
Block a user