优化进度显示

app.py

@@ -123,9 +123,10 @@ def generate(
         progress(progress_value, desc=f"Image generating, {step + 1}/{num_inference_steps} steps")
         return callback_kwargs
     
+    optimizing_steps = int(num_inference_steps * upscaler_strength)
     def callback2(pipe, step, timestep, callback_kwargs):
-        progress_value = 0.6 + ((step+1.0)/num_inference_steps)*(0.4/1.0)
-        progress(progress_value, desc=f"Image optimizing, {step + 1}/{num_inference_steps} steps")
+        progress_value = 0.6 + ((step+1.0)/optimizing_steps)*(0.4/1.0)
+        progress(progress_value, desc=f"Image optimizing, {step + 1}/{optimizing_steps} steps")
         return callback_kwargs
 
     try:
@@ -163,7 +164,6 @@ def generate(
                 output_type="latent",
                 callback_on_step_end=callback1
             ).images
-        progress(0.6,desc="Image optimizing")
         upscaled_latents = utils.upscale(latents, "nearest-exact", upscale_by)
         images = upscaler_pipe(
                 prompt=prompt,