linter

facebookresearch · Dec 15, 2024 · 14f99df · 14f99df
1 parent 34e0f7d
commit 14f99df
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/examples/image/training/eval_loop.py b/examples/image/training/eval_loop.py
@@ -95,7 +95,10 @@ 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, source_distribution_p=p
+            model=cfg_scaled_model,
+            path=path,
+            vocabulary_size=257,
+            source_distribution_p=p,
         )
     else:
         solver = ODESolver(velocity_model=cfg_scaled_model)