Spaces:
Running
on
Zero
Running
on
Zero
test
Browse files- app.py +1 -1
- 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)
|