chore(fix): Fix wrong channels in UpBlock2D

oxideai · Apr 13, 2024 · 1ae1ded · 1ae1ded
1 parent 5a80bbb
commit 1ae1ded
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/models/unet_2d_blocks.rs b/src/models/unet_2d_blocks.rs
@@ -658,7 +658,7 @@ impl UpBlock2DConfig {
 
                 let in_channels = resnet_in_channels + res_skip_channels;
 
-                ResnetBlock2DConfig::new(self.in_channels)
+                ResnetBlock2DConfig::new(in_channels)
                     .with_out_channels(Some(self.out_channels))
                     .with_temb_channels(self.temb_channels)
                     .with_eps(self.resnet_eps)