Spaces:

JMalott
/

ai_architecture

Runtime error

JMalott commited on Sep 6, 2022

Commit

3be0579

•

1 Parent(s): 21adb73

Update min_dalle/min_dalle.py

Files changed (1) hide show

min_dalle/min_dalle.py CHANGED Viewed

@@ -243,7 +243,7 @@ class MinDalle:
             #torch.cuda.empty_cache()
             #torch.cpu.empty_cache()
             #with torch.cuda.amp.autocast(dtype=self.dtype):
-            del image_tokens[i + 1], attention_state = self.decoder.forward(
                 settings=settings,
                 attention_mask=attention_mask,
                 encoder_state=encoder_state,

             #torch.cuda.empty_cache()
             #torch.cpu.empty_cache()
             #with torch.cuda.amp.autocast(dtype=self.dtype):
+            image_tokens[i + 1], attention_state = self.decoder.forward(
                 settings=settings,
                 attention_mask=attention_mask,
                 encoder_state=encoder_state,