fix pbr model

This commit is contained in:
Zeqiang Lai
2025-06-14 11:26:32 +08:00
committed by GitHub
parent dfcd5c829d
commit e447d666f6

View File

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