modify

stable_diffusion.py

@@ -912,7 +912,8 @@ class Generate2img(Module):
 
         with torch.no_grad():
             # this is diffusion
-            for index, timestep in (t := tqdm(list(enumerate(timesteps))[::-1])):
+            t = tqdm(list(enumerate(timesteps))[::-1])
+            for index, timestep in t:
                 t.set_description("%3d %3d" % (index, timestep))
                 e_t = self.get_model_latent_output(latent.clone(), timestep, self.model.unet, context.clone(),
                                        unconditional_context.clone())