From dfcd5c829d36d5704cafd479f46bf77f9e489214 Mon Sep 17 00:00:00 2001
From: Zeqiang Lai <916323301@qq.com>
Date: Sat, 14 Jun 2025 10:35:18 +0800
Subject: [PATCH] fix huggingface loading

---
 hy3dpaint/hunyuanpaintpbr/pipeline.py                  | 4 ++--
 hy3dpaint/hunyuanpaintpbr/{ => unet}/attn_processor.py | 0
 hy3dpaint/hunyuanpaintpbr/{ => unet}/model.py          | 3 +--
 hy3dpaint/hunyuanpaintpbr/{ => unet}/modules.py        | 0
 4 files changed, 3 insertions(+), 4 deletions(-)
 rename hy3dpaint/hunyuanpaintpbr/{ => unet}/attn_processor.py (100%)
 rename hy3dpaint/hunyuanpaintpbr/{ => unet}/model.py (99%)
 rename hy3dpaint/hunyuanpaintpbr/{ => unet}/modules.py (100%)

diff --git a/hy3dpaint/hunyuanpaintpbr/pipeline.py b/hy3dpaint/hunyuanpaintpbr/pipeline.py
index 56bb30a..871dae5 100644
--- a/hy3dpaint/hunyuanpaintpbr/pipeline.py
+++ b/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 .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",
diff --git a/hy3dpaint/hunyuanpaintpbr/attn_processor.py b/hy3dpaint/hunyuanpaintpbr/unet/attn_processor.py
similarity index 100%
rename from hy3dpaint/hunyuanpaintpbr/attn_processor.py
rename to hy3dpaint/hunyuanpaintpbr/unet/attn_processor.py
diff --git a/hy3dpaint/hunyuanpaintpbr/model.py b/hy3dpaint/hunyuanpaintpbr/unet/model.py
similarity index 99%
rename from hy3dpaint/hunyuanpaintpbr/model.py
rename to hy3dpaint/hunyuanpaintpbr/unet/model.py
index 212efc8..f98ca7e 100644
--- a/hy3dpaint/hunyuanpaintpbr/model.py
+++ b/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
 
 
diff --git a/hy3dpaint/hunyuanpaintpbr/modules.py b/hy3dpaint/hunyuanpaintpbr/unet/modules.py
similarity index 100%
rename from hy3dpaint/hunyuanpaintpbr/modules.py
rename to hy3dpaint/hunyuanpaintpbr/unet/modules.py