Spaces:
Sleeping
Sleeping
:tada: change some process
Browse files- app.py +2 -0
- requirements.txt +1 -0
app.py
CHANGED
@@ -75,6 +75,7 @@ def inference(
|
|
75 |
|
76 |
generator = torch.Generator(device=device).manual_seed(seed)
|
77 |
|
|
|
78 |
images = pipe(
|
79 |
prompt,
|
80 |
negative_prompt=negative_prompt,
|
@@ -86,6 +87,7 @@ def inference(
|
|
86 |
generator=generator,
|
87 |
).images
|
88 |
|
|
|
89 |
if num_images % 2 == 1:
|
90 |
image = make_image_grid(images, rows=num_images, cols=1)
|
91 |
else:
|
|
|
75 |
|
76 |
generator = torch.Generator(device=device).manual_seed(seed)
|
77 |
|
78 |
+
progress(0.3, 'Generating images...')
|
79 |
images = pipe(
|
80 |
prompt,
|
81 |
negative_prompt=negative_prompt,
|
|
|
87 |
generator=generator,
|
88 |
).images
|
89 |
|
90 |
+
progress(0.9, f'Done generating {num_images} images')
|
91 |
if num_images % 2 == 1:
|
92 |
image = make_image_grid(images, rows=num_images, cols=1)
|
93 |
else:
|
requirements.txt
CHANGED
@@ -11,3 +11,4 @@ spaces
|
|
11 |
Pillow
|
12 |
protobuf
|
13 |
peft
|
|
|
|
11 |
Pillow
|
12 |
protobuf
|
13 |
peft
|
14 |
+
compel
|