fix from_ddpm #12

tqch · Oct 7, 2023 · 9879834 · 9879834
1 parent 5a2566e
commit 9879834
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ddim.py b/ddim.py
@@ -113,7 +113,7 @@ def p_sample(self, denoise_fn, shape, device=torch.device("cpu"), noise=None, se
  return x_t
 
  @classmethod
- def from_ddpm(cls, eta, subsequence):
+ def from_ddpm(cls, diffusion, eta, subsequence):
  return cls(**{
  k: diffusion.__dict__.get(k, None)
  for k in ["betas", "model_mean_type", "model_var_type", "loss_type"]