LL3RD commited on
Commit
a1b34c3
·
1 Parent(s): 4a51e72
Files changed (2) hide show
  1. app.py +1 -1
  2. dreamfuse_inference.py +2 -2
app.py CHANGED
@@ -502,7 +502,7 @@ class DreamFuseGUI:
502
  size_select = gr.Radio(
503
  choices=["512", "768", "1024"],
504
  value="768",
505
- label="Resolution",
506
  )
507
  prompt_text = gr.Textbox(label="Prompt", placeholder="text prompt", value="")
508
  text_strength = gr.Slider(minimum=1, maximum=10, step=1, label="Text Strength", value=1, visible=False)
 
502
  size_select = gr.Radio(
503
  choices=["512", "768", "1024"],
504
  value="768",
505
+ label="Resolution (Higher resolution improves quality, but slows down generation.)",
506
  )
507
  prompt_text = gr.Textbox(label="Prompt", placeholder="text prompt", value="")
508
  text_strength = gr.Slider(minimum=1, maximum=10, step=1, label="Text Strength", value=1, visible=False)
dreamfuse_inference.py CHANGED
@@ -353,8 +353,8 @@ class DreamFuseInference:
353
  trans = {}
354
 
355
  size_select = int(size_select)
356
- if size_select == 1024: text_strength = 5
357
- if size_select == 768: text_strength = 3
358
 
359
  r, g, b, ori_a = foreground_img.split()
360
  fg_img_scale, fg_img = self.transform_foreground_original(foreground_img, background_img, trans)
 
353
  trans = {}
354
 
355
  size_select = int(size_select)
356
+ if size_select == 1024 and prompt != "": text_strength = 5
357
+ if size_select == 768 and prompt != "": text_strength = 3
358
 
359
  r, g, b, ori_a = foreground_img.split()
360
  fg_img_scale, fg_img = self.transform_foreground_original(foreground_img, background_img, trans)