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
|