From a8c6a5911155e14b5b6a3c27af54f390e6f3e5e3 Mon Sep 17 00:00:00 2001 From: Damian Stewart Date: Sat, 17 Jun 2023 09:27:05 +0200 Subject: [PATCH] fix bug with empty samples.txt --- utils/sample_generator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/sample_generator.py b/utils/sample_generator.py index 6607b7b..dd13bf8 100644 --- a/utils/sample_generator.py +++ b/utils/sample_generator.py @@ -176,7 +176,7 @@ class SampleGenerator: wants_random_caption=p.get('random_caption', False) ) for p in sample_requests_config] if len(self.sample_requests) == 0: - self._make_random_caption_sample_requests() + self.sample_requests = self._make_random_caption_sample_requests() @torch.no_grad() def generate_samples(self, pipe: StableDiffusionPipeline, global_step: int):