Spaces:
Sleeping
Sleeping
kumahiyo
commited on
Commit
•
04b3b49
1
Parent(s):
d1482ff
decrease image count
Browse files
main.py
CHANGED
@@ -54,9 +54,9 @@ def draw(data: Data):
|
|
54 |
pipe = pipe.to('cuda')
|
55 |
|
56 |
generator = torch.Generator("cuda").manual_seed(seed)
|
57 |
-
images = pipe(prompt, negative_prompt=n_prompt, guidance_scale=7.5, generator=generator, num_images_per_prompt=
|
58 |
|
59 |
-
grid = image_grid(images, rows=
|
60 |
|
61 |
fileName = "sd_" + str(time.time()) + '.png'
|
62 |
grid.save("/code/tmpdir/" + fileName)
|
|
|
54 |
pipe = pipe.to('cuda')
|
55 |
|
56 |
generator = torch.Generator("cuda").manual_seed(seed)
|
57 |
+
images = pipe(prompt, negative_prompt=n_prompt, guidance_scale=7.5, generator=generator, num_images_per_prompt=2).images
|
58 |
|
59 |
+
grid = image_grid(images, rows=2, cols=1)
|
60 |
|
61 |
fileName = "sd_" + str(time.time()) + '.png'
|
62 |
grid.save("/code/tmpdir/" + fileName)
|