Spaces:

JMalott
/

ai_architecture

Runtime error

JMalott commited on Sep 3, 2022

Commit

3c3d44d

•

1 Parent(s): 319b1b5

Update min_dalle/min_dalle.py

Files changed (1) hide show

min_dalle/min_dalle.py CHANGED Viewed

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