Merge pull request #11 from Zeqiang-Lai/main

fix pbr model huggingface loading

hy3dpaint/hunyuanpaintpbr/__init__.py

@@ -13,14 +13,14 @@
 # by Tencent in accordance with TENCENT HUNYUAN COMMUNITY LICENSE AGREEMENT.
 
 from .pipeline import HunyuanPaintPipeline
-from .model import HunyuanPaint
+from .unet.model import HunyuanPaint
-from .modules import (
+from .unet.modules import (
     Dino_v2,
     Basic2p5DTransformerBlock,
     ImageProjModel,
     UNet2p5DConditionModel,
 )
-from .attn_processor import (
+from .unet.attn_processor import (
     PoseRoPEAttnProcessor2_0,
     SelfAttnProcessor2_0,
     RefAttnProcessor2_0,

hy3dpaint/hunyuanpaintpbr/pipeline.py

@@ -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 .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__ = [
     "HunyuanPaintPipeline",

hy3dpaint/hunyuanpaintpbr/unet/model.py

@@ -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