From 17b730d2b7aba65176ce6168074f29232ef87071 Mon Sep 17 00:00:00 2001 From: Walter Hugo Lopez Pinaya Date: Thu, 12 Jan 2023 17:02:55 +0000 Subject: [PATCH] Remove ununsed predict_epsilon Signed-off-by: Walter Hugo Lopez Pinaya --- generative/networks/schedulers/ddpm.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/generative/networks/schedulers/ddpm.py b/generative/networks/schedulers/ddpm.py index 7e303400..10de445c 100644 --- a/generative/networks/schedulers/ddpm.py +++ b/generative/networks/schedulers/ddpm.py @@ -183,7 +183,6 @@ def step( model_output: torch.Tensor, timestep: int, sample: torch.Tensor, - predict_epsilon=True, generator: Optional[torch.Generator] = None, ) -> Tuple[torch.Tensor, torch.Tensor]: """ @@ -194,7 +193,6 @@ def step( model_output: direct output from learned diffusion model. timestep: current discrete timestep in the diffusion chain. sample: current instance of sample being created by diffusion process. - predict_epsilon: flag to use when model predicts the samples directly instead of the noise, epsilon. generator: random number generator. Returns: