if progress bars are disabled, log a short message instead

This commit is contained in:
Damian Stewart 2023-03-03 10:50:48 +01:00
parent 644c9e6b2a
commit 29c93eca03
1 changed files with 3 additions and 0 deletions

View File

@ -186,6 +186,9 @@ class SampleGenerator:
except:
font = ImageFont.load_default()
if not self.show_progress_bars:
print(f" * Generating samples at gs:{global_step} for {len(self.sample_requests)} prompts")
sample_index = 0
with autocast():
batch: list[SampleRequest]