Spaces:

JMalott
/

ai_architecture

Runtime error

JMalott commited on Sep 6, 2022

Commit

14ae027

1 Parent(s): 3be0579

Update min_dalle/models/dalle_bart_decoder.py

Files changed (1) hide show

min_dalle/models/dalle_bart_decoder.py CHANGED Viewed

@@ -176,4 +176,6 @@ class DalleBartDecoder(nn.Module):
         logits.exp_()
         logits *= is_kept.to(torch.float32)
         image_tokens = torch.multinomial(logits, 1)[:, 0]
         return image_tokens, attention_state

         logits.exp_()
         logits *= is_kept.to(torch.float32)
         image_tokens = torch.multinomial(logits, 1)[:, 0]
+        del logits
+        print("hi")
         return image_tokens, attention_state