Spaces:

kumahiyo
/

line-bot-stable-diffusion

Sleeping

kumahiyo commited on Mar 4, 2023

Commit

4e425e1

•

1 Parent(s): 7e6f588

fix count of images

Files changed (1) hide show

main.py CHANGED Viewed

@@ -54,7 +54,7 @@ def draw(data: Data):
 		pipe = pipe.to('cuda')
 		generator = torch.Generator("cuda").manual_seed(seed)
-		images = pipe(prompt, negative_prompt=n_prompt, guidance_scale=7.5, generator=generator, num_images_per_prompt=3).images
 		grid = image_grid(images, rows=2, cols=2)

 		pipe = pipe.to('cuda')
 		generator = torch.Generator("cuda").manual_seed(seed)
+		images = pipe(prompt, negative_prompt=n_prompt, guidance_scale=7.5, generator=generator, num_images_per_prompt=4).images
 		grid = image_grid(images, rows=2, cols=2)