diff --git a/hy3dpaint/hunyuanpaintpbr/__init__.py b/hy3dpaint/hunyuanpaintpbr/__init__.py
index ede751a..930d2fc 100644
--- a/hy3dpaint/hunyuanpaintpbr/__init__.py
+++ b/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 .modules import (
+from .unet.model import HunyuanPaint
+from .unet.modules import (
     Dino_v2,
     Basic2p5DTransformerBlock,
     ImageProjModel,
     UNet2p5DConditionModel,
 )
-from .attn_processor import (
+from .unet.attn_processor import (
     PoseRoPEAttnProcessor2_0,
     SelfAttnProcessor2_0,
     RefAttnProcessor2_0,
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