Use fast_after_steps only with use_fast_schedule

utils/latents.py

@@ -74,7 +74,10 @@ def compose_latents(model_dict, latents_all_list, mask_tensor_list, num_inferenc
         latents_all, mask_tensor = latents_all_list[mask_idx], mask_tensor_list[mask_idx]
         foreground_indices = foreground_indices * (~mask_tensor) + (mask_idx + 1) * mask_tensor
         mask_tensor_expanded = mask_tensor[None, None, None, ...].to(dtype)
-        composed_latents = composed_latents * (1. - mask_tensor_expanded) + latents_all[:fast_after_steps + 1] * mask_tensor_expanded
+        if use_fast_schedule:
+            composed_latents = composed_latents * (1. - mask_tensor_expanded) + latents_all[:fast_after_steps + 1] * mask_tensor_expanded
+        else:
+            composed_latents = composed_latents * (1. - mask_tensor_expanded) + latents_all * mask_tensor_expanded
         
     composed_latents, foreground_indices = composed_latents.to(torch_device), foreground_indices.to(torch_device)
     return composed_latents, foreground_indices