Spaces:

JMalott
/

ai_architecture

Runtime error

JMalott commited on Sep 3, 2022

Commit

71b8a97

•

1 Parent(s): 478d137

Update min_dalle/min_dalle.py

Files changed (1) hide show

min_dalle/min_dalle.py CHANGED Viewed

@@ -234,7 +234,7 @@ class MinDalle:
             dtype=torch.float32,
             device=self.device
         )
-        for i in range(round(IMAGE_TOKEN_COUNT/2) ):
             if(st.session_state.page != 0):
                 break
             st.session_state.bar.progress(i/IMAGE_TOKEN_COUNT)

             dtype=torch.float32,
             device=self.device
         )
+        for i in range( IMAGE_TOKEN_COUNT ):
             if(st.session_state.page != 0):
                 break
             st.session_state.bar.progress(i/IMAGE_TOKEN_COUNT)