mrq
/
DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

uh

pull/2/head
James Betker 2022-07-18 17:15:27 +07:00
parent c959e530cb
commit 8d7692c1e0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/trainer/injectors/gaussian_diffusion_injector.py
Unescape Escape View File

@ -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: