t2m

2026-04-24 15:06:17 +00:00 · 2026-04-21 20:12:15 +08:00
parent 95cfb77881
commit f5a3201d42
9 changed files with 133 additions and 369 deletions
--- a/diffsynth/pipelines/ace_step.py
+++ b/diffsynth/pipelines/ace_step.py
@@ -145,7 +145,6 @@ class AceStepPipeline(BasePipeline):
        # 4. Denoise loop
        self.load_models_to_device(self.in_iteration_models)
        models = {name: getattr(self, name) for name in self.in_iteration_models}
-        self.momentum_buffer = MomentumBuffer()
        for progress_id, timestep in enumerate(progress_bar_cmd(self.scheduler.timesteps)):
            timestep = timestep.to(dtype=self.torch_dtype, device=self.device)
            noise_pred = self.cfg_guided_model_fn(