Mirrors
/
EveryDream2trainer
mirror of https://github.com/victorchall/EveryDream2trainer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix bug with empty samples.txt

This commit is contained in:
Damian Stewart 2023-06-17 09:27:05 +02:00
parent 403f7ddf07
commit a8c6a59111
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/sample_generator.py
View File

@ -176,7 +176,7 @@ class SampleGenerator:
wants_random_caption=p.get('random_caption', False) wants_random_caption=p.get('random_caption', False)
) for p in sample_requests_config] ) for p in sample_requests_config]
if len(self.sample_requests) == 0: if len(self.sample_requests) == 0:
self._make_random_caption_sample_requests() self.sample_requests = self._make_random_caption_sample_requests()
@torch.no_grad() @torch.no_grad()
def generate_samples(self, pipe: StableDiffusionPipeline, global_step: int): def generate_samples(self, pipe: StableDiffusionPipeline, global_step: int):