Merge pull request #14 from ashaw596/patch-1

Fix got an unexpected keyword argument in examples/image/training/eval_loop.py in MixtureDiscreteEulerSolver.__init__
facebookresearch · Dec 15, 2024 · 34e0f7d · 34e0f7d
2 parents 885d166 + 3ce3e32
commit 34e0f7d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)