Spaces:

Zhengyi
/

CRM

Running on Zero

Zhengyi commited on Mar 11

Commit

a71660a

•

1 Parent(s): bf7d47c

RM exformers in unet

Files changed (1) hide show

model/archs/unet.py CHANGED Viewed

@@ -40,7 +40,7 @@ class UNetPP(nn.Module):
                 ),
             )
-        self.unet.enable_xformers_memory_efficient_attention()
         if in_channels > 12:
             self.learned_plane = torch.nn.parameter.Parameter(torch.zeros([1,in_channels-12,256,256*3]))

                 ),
             )
+        # self.unet.enable_xformers_memory_efficient_attention()
         if in_channels > 12:
             self.learned_plane = torch.nn.parameter.Parameter(torch.zeros([1,in_channels-12,256,256*3]))