Upload block.py with huggingface_hub

block.py

@@ -9,6 +9,7 @@ References:
 - Modular Diffusers: https://huggingface.co/docs/diffusers/main/en/modular_diffusers
 """
 
+
 # ============================================================
 # utils_tiling
 # ============================================================
@@ -2229,16 +2230,24 @@ class UltimateSDUpscaleMultiDiffusionStep(ModularPipelineBlocks):
         image_latents = enc_state.get("image_latents")
 
         # --- Re-compute timesteps for this pass's strength ---
-        # Reset scheduler timesteps so the current strength is applied correctly.
+        # The outer set_timesteps block overwrites num_inference_steps in state with
+        # a strength-truncated value. To avoid double-truncation, we recover the
+        # original step count: truncated = original - floor(original * strength),
+        # so original = ceil(truncated / (1 - strength)). We clamp to at least 1.
         pass_strength = block_state._current_pass_strength
+        truncated_steps = block_state.num_inference_steps
+        if pass_strength < 1.0:
+            original_steps = max(1, int(math.ceil(truncated_steps / (1.0 - pass_strength))))
+        else:
+            original_steps = truncated_steps
         _ts, _nsteps = retrieve_timesteps(
             components.scheduler,
-            block_state.num_inference_steps,
+            original_steps,
             components._execution_device,
         )
         from diffusers.modular_pipelines.stable_diffusion_xl.before_denoise import StableDiffusionXLImg2ImgSetTimestepsStep
         timesteps, num_inf_steps = StableDiffusionXLImg2ImgSetTimestepsStep.get_timesteps(
-            components, block_state.num_inference_steps, pass_strength, components._execution_device,
+            components, original_steps, pass_strength, components._execution_device,
         )
         latent_timestep = timesteps[:1].repeat(block_state.batch_size * block_state.num_images_per_prompt)
 
@@ -2572,7 +2581,7 @@ class UltimateSDUpscaleMultiDiffusionStep(ModularPipelineBlocks):
 
 
 # ============================================================
-# modular_blocks
+# modular_blocks_ultimate_sd_upscale
 # ============================================================
 
 # Copyright 2025 The HuggingFace Team. All rights reserved.
@@ -2779,3 +2788,4 @@ class UltimateSDUpscaleModularPipeline(StableDiffusionXLModularPipeline):
     """
 
     default_blocks_name = "UltimateSDUpscaleBlocks"
+