From 14f99df2f12c7add6e1ad1b203953fff39a950d3 Mon Sep 17 00:00:00 2001 From: Itai Gat Date: Sun, 15 Dec 2024 20:54:55 +0000 Subject: [PATCH] linter --- examples/image/training/eval_loop.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/examples/image/training/eval_loop.py b/examples/image/training/eval_loop.py index 1715925..f39220d 100644 --- 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)