Update min_dalle/min_dalle.py

min_dalle/min_dalle.py

@@ -238,8 +238,6 @@ class MinDalle:
             if(st.session_state.page != 0):
                 break
             st.session_state.bar.progress(i/IMAGE_TOKEN_COUNT)
-            
-            print(i)
 
             torch.cuda.empty_cache()                
             with torch.cuda.amp.autocast(dtype=self.dtype):
@@ -252,8 +250,8 @@ class MinDalle:
                     token_index=token_indices[[i]]
                 )
 
-            with torch.cuda.amp.autocast(dtype=torch.float32):
-                if ((i + 1) % 32 == 0 and progressive_outputs) or i + 1 == 256:
+            with torch.cuda.amp.autocast(dtype=torch.float16):
+                if ((i + 1) % 16 == 0 and progressive_outputs) or i + 1 == 256:
                     yield self.image_grid_from_tokens(
                         image_tokens=image_tokens[1:].T,
                         is_seamless=is_seamless,