Update min_dalle/models/dalle_bart_decoder.py

min_dalle/models/dalle_bart_decoder.py

@@ -161,7 +161,7 @@ class DalleBartDecoder(nn.Module):
             )
         decoder_state = self.final_ln(decoder_state)
         logits = self.lm_head(decoder_state)
-        del decorder_state
+        del decoder_state 
         temperature = settings[[0]]
         top_k = settings[[1]].to(torch.long)
         supercondition_factor = settings[[2]]
@@ -176,6 +176,7 @@ class DalleBartDecoder(nn.Module):
         logits -= logits_sorted[:, [0]]
         del logits_sorted
         logits /= temperature
+        del temperature
         logits.exp_()
         logits *= is_kept.to(torch.float32)
         image_tokens = torch.multinomial(logits, 1)[:, 0]