Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ from diffusers_patches import pixart_sigma_init_patched_inputs, PixArtSigmaPipel
|
|
19 |
DESCRIPTION = """ # Instant Image
|
20 |
### Super fast text to Image Generator.
|
21 |
### <span style='color: red;'>You may change the steps from 9 to 15, if you didn't get satisfied results.
|
22 |
-
### First Image processing takes time
|
23 |
"""
|
24 |
if not torch.cuda.is_available():
|
25 |
DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"
|
@@ -155,7 +155,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
|
|
155 |
|
156 |
@torch.no_grad()
|
157 |
@torch.inference_mode()
|
158 |
-
@spaces.GPU(duration=
|
159 |
def generate(
|
160 |
prompt: str,
|
161 |
negative_prompt: str = "",
|
@@ -163,8 +163,8 @@ def generate(
|
|
163 |
use_negative_prompt: bool = False,
|
164 |
num_imgs: int = 1,
|
165 |
seed: int = 0,
|
166 |
-
width: int =
|
167 |
-
height: int =
|
168 |
schedule: str = 'DPM-Solver',
|
169 |
dpms_guidance_scale: float = 3.5,
|
170 |
dpms_inference_steps: int = 9,
|
@@ -287,14 +287,14 @@ with gr.Blocks(css="style.css") as demo:
|
|
287 |
minimum=256,
|
288 |
maximum=MAX_IMAGE_SIZE,
|
289 |
step=32,
|
290 |
-
value=
|
291 |
)
|
292 |
height = gr.Slider(
|
293 |
label="Height",
|
294 |
minimum=256,
|
295 |
maximum=MAX_IMAGE_SIZE,
|
296 |
step=32,
|
297 |
-
value=
|
298 |
)
|
299 |
with gr.Row():
|
300 |
dpms_guidance_scale = gr.Slider(
|
|
|
19 |
DESCRIPTION = """ # Instant Image
|
20 |
### Super fast text to Image Generator.
|
21 |
### <span style='color: red;'>You may change the steps from 9 to 15, if you didn't get satisfied results.
|
22 |
+
### First Image processing takes time then images generate faster.
|
23 |
"""
|
24 |
if not torch.cuda.is_available():
|
25 |
DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"
|
|
|
155 |
|
156 |
@torch.no_grad()
|
157 |
@torch.inference_mode()
|
158 |
+
@spaces.GPU(duration=30)
|
159 |
def generate(
|
160 |
prompt: str,
|
161 |
negative_prompt: str = "",
|
|
|
163 |
use_negative_prompt: bool = False,
|
164 |
num_imgs: int = 1,
|
165 |
seed: int = 0,
|
166 |
+
width: int = 2560,
|
167 |
+
height: int = 2560,
|
168 |
schedule: str = 'DPM-Solver',
|
169 |
dpms_guidance_scale: float = 3.5,
|
170 |
dpms_inference_steps: int = 9,
|
|
|
287 |
minimum=256,
|
288 |
maximum=MAX_IMAGE_SIZE,
|
289 |
step=32,
|
290 |
+
value=2560,
|
291 |
)
|
292 |
height = gr.Slider(
|
293 |
label="Height",
|
294 |
minimum=256,
|
295 |
maximum=MAX_IMAGE_SIZE,
|
296 |
step=32,
|
297 |
+
value=2560,
|
298 |
)
|
299 |
with gr.Row():
|
300 |
dpms_guidance_scale = gr.Slider(
|