fbnnb commited on
Commit
2c6dff3
1 Parent(s): 29cf59d

Update lvdm/models/ddpm3d.py

Browse files
Files changed (1) hide show
  1. lvdm/models/ddpm3d.py +1 -1
lvdm/models/ddpm3d.py CHANGED
@@ -488,7 +488,7 @@ class LatentDiffusion(DDPM):
488
  # added
489
  logdir=None,
490
  rand_cond_frame=False,
491
- en_and_decode_n_samples_a_time=8,
492
  *args, **kwargs):
493
  self.num_timesteps_cond = default(num_timesteps_cond, 1)
494
  self.scale_by_std = scale_by_std
 
488
  # added
489
  logdir=None,
490
  rand_cond_frame=False,
491
+ en_and_decode_n_samples_a_time=16,
492
  *args, **kwargs):
493
  self.num_timesteps_cond = default(num_timesteps_cond, 1)
494
  self.scale_by_std = scale_by_std