Update lvdm/models/ddpm3d.py
Browse files- 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=
|
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
|