aljaz/generative-models
1
0
Fork 0
mirror of https://github.com/Stability-AI/generative-models.git synced 2025-12-19 22:34:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix crashing line in logging in sgm/models/diffusion.py (#64)

Browse Source
This commit is contained in:
ablattmann
2023-07-26 10:13:42 +02:00
committed by GitHub
parent 57862fb4c7
commit 68f3f89bd3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
sgm/models/diffusion.py
View File

@@ -307,7 +307,7 @@ class DiffusionEngine(pl.LightningModule):
log["inputs"] = x
z = self.encode_first_stage(x)
log["reconstructions"] = self.decode_first_stage(z)
logger.update(self.log_conditionings(batch, N))
log.update(self.log_conditionings(batch, N))
for k in c:
if isinstance(c[k], torch.Tensor):