Spaces:
Runtime error
Runtime error
Update lvdm/models/ddpm3d.py
Browse files- lvdm/models/ddpm3d.py +2 -0
lvdm/models/ddpm3d.py
CHANGED
|
@@ -186,6 +186,8 @@ class DDPM(pl.LightningModule):
|
|
| 186 |
self.register_buffer('lvlb_weights', lvlb_weights, persistent=False)
|
| 187 |
assert not torch.isnan(self.lvlb_weights).all()
|
| 188 |
|
|
|
|
|
|
|
| 189 |
@contextmanager
|
| 190 |
def ema_scope(self, context=None):
|
| 191 |
if self.use_ema:
|
|
|
|
| 186 |
self.register_buffer('lvlb_weights', lvlb_weights, persistent=False)
|
| 187 |
assert not torch.isnan(self.lvlb_weights).all()
|
| 188 |
|
| 189 |
+
self.control_scale = 0.6
|
| 190 |
+
|
| 191 |
@contextmanager
|
| 192 |
def ema_scope(self, context=None):
|
| 193 |
if self.use_ema:
|