Spaces:

JMalott
/

ai_architecture

Runtime error

JMalott commited on Sep 7, 2022

Commit

c8787d9

•

1 Parent(s): 0adf088

Update min_dalle/models/dalle_bart_decoder.py

Files changed (1) hide show

min_dalle/models/dalle_bart_decoder.py CHANGED Viewed

@@ -162,7 +162,7 @@ class DalleBartDecoder(nn.Module):
         print(tracemalloc.get_traced_memory())
         for i in range(self.layer_count):
-            del decoder_state, attention_state[i] = self.layers[i].forward(
                 decoder_state,
                 encoder_state,
                 attention_state[i],

         print(tracemalloc.get_traced_memory())
         for i in range(self.layer_count):
+            decoder_state, attention_state[i] = self.layers[i].forward(
                 decoder_state,
                 encoder_state,
                 attention_state[i],