Spaces:
Running
on
A10G
Running
on
A10G
JingyeChen
commited on
Commit
•
401dece
1
Parent(s):
93f708e
update
Browse files
app.py
CHANGED
@@ -386,9 +386,9 @@ def text_to_image(prompt,keywords,radio,slider_step,slider_guidance,slider_batch
|
|
386 |
prompt=user_prompt,
|
387 |
generator=generator,
|
388 |
# negative_prompt=negative_prompt,
|
389 |
-
num_inference_steps=
|
390 |
guidance_scale=1,
|
391 |
-
num_images_per_prompt=slider_batch,
|
392 |
).images
|
393 |
os.system('nvidia-smi')
|
394 |
torch.cuda.empty_cache()
|
@@ -400,7 +400,7 @@ with gr.Blocks() as demo:
|
|
400 |
gr.HTML(
|
401 |
"""
|
402 |
<div style="text-align: center; max-width: 1600px; margin: 20px auto;">
|
403 |
-
<h2 style="font-weight: 900; font-size: 2.
|
404 |
TextDiffuser-2: Unleashing the Power of Language Models for Text Rendering
|
405 |
</h2>
|
406 |
<h2 style="font-weight: 460; font-size: 1.1rem; margin: 0rem">
|
|
|
386 |
prompt=user_prompt,
|
387 |
generator=generator,
|
388 |
# negative_prompt=negative_prompt,
|
389 |
+
num_inference_steps=slider_step,
|
390 |
guidance_scale=1,
|
391 |
+
# num_images_per_prompt=slider_batch,
|
392 |
).images
|
393 |
os.system('nvidia-smi')
|
394 |
torch.cuda.empty_cache()
|
|
|
400 |
gr.HTML(
|
401 |
"""
|
402 |
<div style="text-align: center; max-width: 1600px; margin: 20px auto;">
|
403 |
+
<h2 style="font-weight: 900; font-size: 2.4rem; margin: 0rem">
|
404 |
TextDiffuser-2: Unleashing the Power of Language Models for Text Rendering
|
405 |
</h2>
|
406 |
<h2 style="font-weight: 460; font-size: 1.1rem; margin: 0rem">
|