From 8d7692c1e08698d0aa54c056b577c8928fd0eb68 Mon Sep 17 00:00:00 2001 From: James Betker Date: Mon, 18 Jul 2022 17:15:27 -0600 Subject: [PATCH] uh --- codes/trainer/injectors/gaussian_diffusion_injector.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/trainer/injectors/gaussian_diffusion_injector.py b/codes/trainer/injectors/gaussian_diffusion_injector.py index 931f8fad..b31259a0 100644 --- a/codes/trainer/injectors/gaussian_diffusion_injector.py +++ b/codes/trainer/injectors/gaussian_diffusion_injector.py @@ -90,7 +90,7 @@ class GaussianDiffusionInjector(Injector): self.deterministic_sampler.reset() # Keep this reset whenever it is not being used, so it is ready to use automatically. model_inputs = {k: state[v] if isinstance(v, str) else v for k, v in self.model_input_keys.items()} if self.preprocess_fn is not None: - hq = getattr(gen, self.preprocess_fn)(hq, **model_inputs) + hq = getattr(gen.module, self.preprocess_fn)(hq, **model_inputs) t, weights = sampler.sample(hq.shape[0], hq.device) if self.causal_mode: