change some problem 3

2025-11-20 03:04:05 +08:00
parent 0a45856b14
commit ebda75fa5e
1 changed files with 1 additions and 2 deletions
--- a/tools/diffusion/ic_layout_diffusion_optimized.py
+++ b/tools/diffusion/ic_layout_diffusion_optimized.py
@@ -233,9 +233,8 @@ class ManhattanAwareUNet(nn.Module):
            nn.Conv2d(32, out_channels, 3, padding=1)
        )

-        # 时间融合层
+        # 时间融合层 - 与编码器输出通道数匹配
        self.time_fusion = nn.ModuleList([
-            nn.Linear(time_dim, 64),
            nn.Linear(time_dim, 128),
            nn.Linear(time_dim, 256),
            nn.Linear(time_dim, 512),