Fix got an unexpected keyword argument in examples/image/training/eva…

…l_loop.py in MixtureDiscreteEulerSolver.__init__
facebookresearch · Dec 15, 2024 · 3ce3e32 · 3ce3e32
1 parent 885d166
commit 3ce3e32
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)