Skip to content

GitLab

"docs/source/en/vscode:/vscode.git/clone" did not exist on "0ab6dc0f231d1bc712454331ad16a5148cc335af"
Unverified Commit cc2c4ae7 authored by Pu Cao's avatar Pu Cao Committed by GitHub
Browse files

fix inference in custom diffusion (#5329) 



* Update train_custom_diffusion.py

* make style

* Empty-Commit

---------
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
parent 6bd55b54
Show whitespace changes
Inline Side-by-side
Showing with 44 additions and 42 deletions
examples/custom_diffusion/train_custom_diffusion.py
View file @ cc2c4ae7
...@@ -1238,7 +1238,9 @@ def main(args): ...@@ -1238,7 +1238,9 @@ def main(args):
# run inference # run inference
generator = torch.Generator(device=accelerator.device).manual_seed(args.seed) generator = torch.Generator(device=accelerator.device).manual_seed(args.seed)
images = [ images = [
pipeline(args.validation_prompt, num_inference_steps=25, generator=generator, eta=1.0).images[0] pipeline(args.validation_prompt, num_inference_steps=25, generator=generator, eta=1.0).images[
0
]
for _ in range(args.num_validation_images) for _ in range(args.num_validation_images)
] ]
......
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