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.image_processor import PipelineImageInput
|
||||
from diffusers.pipelines.stable_diffusion.pipeline_output import StableDiffusionPipelineOutput
|
||||
from .modules import UNet2p5DConditionModel
|
||||
from .attn_processor import SelfAttnProcessor2_0, RefAttnProcessor2_0, PoseRoPEAttnProcessor2_0
|
||||
from .unet.modules import UNet2p5DConditionModel
|
||||
from .unet.attn_processor import SelfAttnProcessor2_0, RefAttnProcessor2_0, PoseRoPEAttnProcessor2_0
|
||||
|
||||
__all__ = [
|
||||
"HunyuanPaintPipeline",
|
||||
|
||||
@@ -32,9 +32,8 @@ from diffusers import (
|
||||
UNet2DConditionModel,
|
||||
ControlNetModel,
|
||||
)
|
||||
from .pipeline import UNet2p5DConditionModel
|
||||
|
||||
from .modules import Dino_v2
|
||||
from .modules import Dino_v2, UNet2p5DConditionModel
|
||||
import math
|
||||
|
||||
|
||||
Reference in New Issue
Block a user