Spaces:

JMalott
/

ai_architecture

Runtime error

JMalott commited on Aug 18, 2022

Commit

cc89b51

•

1 Parent(s): f533b7c

Update min_dalle/min_dalle.py

Files changed (1) hide show

min_dalle/min_dalle.py CHANGED Viewed

@@ -238,6 +238,8 @@ class MinDalle:
             if(st.session_state.page != 0):
                 break
             st.session_state.bar.progress(i/IMAGE_TOKEN_COUNT)
             torch.cuda.empty_cache()
             with torch.cuda.amp.autocast(dtype=self.dtype):

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