Spaces:

JMalott
/

ai_architecture

Runtime error

JMalott commited on Sep 4, 2022

Commit

7f311ef

•

1 Parent(s): 8e71a67

Update min_dalle/min_dalle.py

Files changed (1) hide show

min_dalle/min_dalle.py CHANGED Viewed

@@ -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)
             torch.cuda.empty_cache()

         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()