Spaces:

Purple11
/

Grounded-Diffusion

Runtime error

Purple11 commited on Feb 17, 2023

Commit

4906d2d

•

1 Parent(s): 2e34ed1

Update ldm/models/seg_module.py

Files changed (1) hide show

ldm/models/seg_module.py CHANGED Viewed

@@ -94,7 +94,7 @@ class TransformerDecoderLayer(nn.Module):
     def forward(self, tgt, memory, pos = None, query_pos = None):
         tgt2 = self.norm1(tgt)
         q = k = self.with_pos_embed(tgt2, query_pos)
-        print('q:',q.shape)
         tgt2 = self.self_attn(q, k, value=tgt2)[0]
         tgt = tgt + self.dropout1(tgt2)

     def forward(self, tgt, memory, pos = None, query_pos = None):
         tgt2 = self.norm1(tgt)
         q = k = self.with_pos_embed(tgt2, query_pos)
+        # print('q:',q.shape)
         tgt2 = self.self_attn(q, k, value=tgt2)[0]
         tgt = tgt + self.dropout1(tgt2)