Update min_dalle/min_dalle.py

min_dalle/min_dalle.py

@@ -237,13 +237,13 @@ class MinDalle:
         for i in range(IMAGE_TOKEN_COUNT):
             if(st.session_state.page != 0):
                 break
+         
             st.session_state.bar.progress(i/IMAGE_TOKEN_COUNT)
-            
       
             #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(
+            del image_tokens[i + 1], attention_state = self.decoder.forward(
                 settings=settings,
                 attention_mask=attention_mask,
                 encoder_state=encoder_state,
@@ -276,8 +276,6 @@ class MinDalle:
             image = image.transpose(1, 0)
             image = image.reshape([grid_size ** 2, 2 ** 8, 2 ** 8, 3])
             yield image
-            del image
-
 
     def generate_image(self, *args, **kwargs) -> Image.Image:
         image_stream = self.generate_image_stream(