Upload handler.py

handler.py

@@ -703,9 +703,18 @@ class EndpointHandler:
                             first_pass_steps = max(1, int(config.num_inference_steps * 0.7))
                             second_pass_steps = max(1, config.num_inference_steps - first_pass_steps)
                             
-                            # Generate valid timesteps for each pass
+                            # Generate valid timesteps for first pass (starts at 1.0)
                             first_pass["timesteps"] = generate_valid_timesteps(first_pass_steps, ALLOWED_TIMESTEPS)
-                            second_pass["timesteps"] = generate_valid_timesteps(second_pass_steps, ALLOWED_TIMESTEPS)
+                            
+                            # Second pass should start at high noise level (like 0.9094) but NOT 1.0
+                            # Find a good starting point in the allowed timesteps (around 0.9094)
+                            start_idx = 5  # This is 0.9094 in ALLOWED_TIMESTEPS
+                            if second_pass_steps == 1:
+                                second_pass["timesteps"] = [ALLOWED_TIMESTEPS[start_idx]]  # Just 0.9094
+                            else:
+                                # Start from 0.9094 and go down, taking second_pass_steps timesteps
+                                end_idx = min(len(ALLOWED_TIMESTEPS), start_idx + second_pass_steps)
+                                second_pass["timesteps"] = ALLOWED_TIMESTEPS[start_idx:end_idx]
                     else:
                         # Case 3: Use default optimized timesteps
                         first_pass["timesteps"] = DEFAULT_FIRST_PASS_TIMESTEPS