Spaces:

JMalott
/

ai_architecture

Runtime error

JMalott commited on Sep 6, 2022

Commit

27df13b

1 Parent(s): a5c89d0

Update min_dalle/min_dalle.py

Files changed (1) hide show

min_dalle/min_dalle.py CHANGED Viewed

@@ -250,7 +250,7 @@ class MinDalle:
                     prev_tokens=image_tokens[i],
                     token_index=token_indices[[i]]
                 )
-                del attention_state
             with torch.cuda.amp.autocast(dtype=torch.float16):
                 if ((i + 1) % 32 == 0 and progressive_outputs) or i + 1 == 256:

                     prev_tokens=image_tokens[i],
                     token_index=token_indices[[i]]
                 )
             with torch.cuda.amp.autocast(dtype=torch.float16):
                 if ((i + 1) % 32 == 0 and progressive_outputs) or i + 1 == 256: