Skip to content

GitLab

Unverified Commit cd9d0913 authored by Alex McKinney's avatar Alex McKinney Committed by GitHub
Browse files

Fixes eval generator init in `train_text_to_image_lora.py` (#3678)

parent fdec2318
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 2 deletions
examples/text_to_image/train_text_to_image_lora.py
View file @ cd9d0913
...@@ -835,7 +835,9 @@ def main(): ...@@ -835,7 +835,9 @@ def main():
pipeline.set_progress_bar_config(disable=True) pipeline.set_progress_bar_config(disable=True)
# run inference # run inference
generator = torch.Generator(device=accelerator.device).manual_seed(args.seed) generator = torch.Generator(device=accelerator.device)
if args.seed is not None:
generator = generator.manual_seed(args.seed)
images = [] images = []
for _ in range(args.num_validation_images): for _ in range(args.num_validation_images):
images.append( images.append(
...@@ -891,7 +893,9 @@ def main(): ...@@ -891,7 +893,9 @@ def main():
pipeline.unet.load_attn_procs(args.output_dir) pipeline.unet.load_attn_procs(args.output_dir)
# run inference # run inference
generator = torch.Generator(device=accelerator.device).manual_seed(args.seed) generator = torch.Generator(device=accelerator.device)
if args.seed is not None:
generator = generator.manual_seed(args.seed)
images = [] images = []
for _ in range(args.num_validation_images): for _ in range(args.num_validation_images):
images.append(pipeline(args.validation_prompt, num_inference_steps=30, generator=generator).images[0]) images.append(pipeline(args.validation_prompt, num_inference_steps=30, generator=generator).images[0])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment