r/stable-diffusion-finetune
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

prevent sampling in first steps

Browse source
This commit is contained in:
Patrick Esser 2022-07-22 12:35:39 +00:00
parent 6dfe59e9b0
commit 9be5028617
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
main.py
View file

@ -350,6 +350,7 @@ class ImageLogger(Callback):
if (self.check_frequency(check_idx) and # batch_idx % self.batch_freq == 0 if (self.check_frequency(check_idx) and # batch_idx % self.batch_freq == 0
hasattr(pl_module, "log_images") and hasattr(pl_module, "log_images") and
callable(pl_module.log_images) and callable(pl_module.log_images) and
batch_idx > 5 and
self.max_images > 0): self.max_images > 0):
logger = type(pl_module.logger) logger = type(pl_module.logger)