From 3ce3e32c26bf2405e21363debc7cfa2860d1c686 Mon Sep 17 00:00:00 2001 From: ashaw596 Date: Sun, 15 Dec 2024 10:04:35 -0800 Subject: [PATCH] Fix got an unexpected keyword argument in examples/image/training/eval_loop.py in MixtureDiscreteEulerSolver.__init__ --- examples/image/training/eval_loop.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/image/training/eval_loop.py b/examples/image/training/eval_loop.py index 527b62b..1715925 100644 --- a/examples/image/training/eval_loop.py +++ b/examples/image/training/eval_loop.py @@ -95,7 +95,7 @@ def eval_model( p = torch.zeros(size=[257], dtype=torch.float32, device=device) p[256] = 1.0 solver = MixtureDiscreteEulerSolver( - model=cfg_scaled_model, path=path, vocabulary_size=257, p=p + model=cfg_scaled_model, path=path, vocabulary_size=257, source_distribution_p=p ) else: solver = ODESolver(velocity_model=cfg_scaled_model)