Spaces:

xco2
/

small_diffusion

Running

xco2 commited on Sep 25, 2023

Commit

687cb7c

1 Parent(s): ebfe12f

init

Files changed (1) hide show

net/UNet.py CHANGED Viewed

@@ -202,7 +202,7 @@ class EncoderBlock(nn.Module):
         skip = self.conv_skip(x)
         for i, layer in enumerate(self.model_list):
-            x = layer(x)
             if i == 0:
                 x = self.att_block(x)
             if i < self.model_list_len - 1:

         skip = self.conv_skip(x)
         for i, layer in enumerate(self.model_list):
+            x = layer(x)
             if i == 0:
                 x = self.att_block(x)
             if i < self.model_list_len - 1: