Update min_dalle/min_dalle.py

min_dalle/min_dalle.py

@@ -238,7 +238,7 @@ 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)
+            #st.session_state.bar.progress(i/IMAGE_TOKEN_COUNT)
 
             torch.cuda.empty_cache()                
             
@@ -254,7 +254,7 @@ class MinDalle:
                     token_index=token_indices[[i]]
                 )
                 
-            torch.cuda.empty_cache()
+            #torch.cuda.empty_cache()
 
             with torch.cuda.amp.autocast(dtype=torch.float16):
                 if ((i + 1) % 16 == 0 and progressive_outputs) or i + 1 == 256: