JingyeChen commited on
Commit
401dece
1 Parent(s): 93f708e
Files changed (1) hide show
  1. app.py +3 -3
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=1,
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.5rem; 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">
 
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">