Update min_dalle/min_dalle.py

min_dalle/min_dalle.py

@@ -242,9 +242,9 @@ 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)
+            #if(st.session_state.page != 0):
+            #    break
+            #st.session_state.bar.progress(i/IMAGE_TOKEN_COUNT)
 
             torch.cuda.empty_cache()                
             
@@ -268,6 +268,7 @@ class MinDalle:
                         is_seamless=is_seamless,
                         is_verbose=is_verbose
                     )
+                    
             # displaying the memory
             print(tracemalloc.get_traced_memory())