fix bug with empty samples.txt

2023-06-17 09:27:05 +02:00 · 2023-06-17 09:27:05 +02:00 · a8c6a59111
parent 403f7ddf07
commit a8c6a59111
1 changed files with 1 additions and 1 deletions
--- 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):