fix bugs(admas timedeltas)

2025-05-20 12:27:34 +08:00
parent 699d89a772
commit 3693640ca3
2 changed files with 2 additions and 2 deletions
--- a/src/diffusion/stateful_flow_matching/adam_sampling.py
+++ b/src/diffusion/stateful_flow_matching/adam_sampling.py
@@ -98,7 +98,7 @@ class AdamLMSampler(BaseSampler):
            if i % self.state_refresh_rate == 0:
                state = None
            out, state = net(cfg_x, cfg_t, cfg_condition, state)
-            out = self.guidance_fn(out, self.guidances[i])
+            out = self.guidance_fn(out, self.guidance)
            pred_trajectory.append(out)
            out = torch.zeros_like(out)
            order = len(self.solver_coeffs[i])