Spaces:

JMalott
/

ai_architecture

Runtime error

JMalott commited on Sep 7, 2022

Commit

8190b2f

1 Parent(s): dbc2880

Update min_dalle/models/dalle_bart_decoder.py

Files changed (1) hide show

min_dalle/models/dalle_bart_decoder.py CHANGED Viewed

@@ -169,6 +169,7 @@ class DalleBartDecoder(nn.Module):
                 attention_mask,
                 token_index
             )
         print(tracemalloc.get_traced_memory())
         decoder_state = self.final_ln(decoder_state)
         logits = self.lm_head(decoder_state)

                 attention_mask,
                 token_index
             )
+            del decoder_state
         print(tracemalloc.get_traced_memory())
         decoder_state = self.final_ln(decoder_state)
         logits = self.lm_head(decoder_state)