Spaces:
Running
Running
Commit
·
40d7c7f
1
Parent(s):
30b02ae
adjust
Browse files
app.py
CHANGED
|
@@ -140,10 +140,8 @@ def control_ui():
|
|
| 140 |
with gr.Tab('局部重绘', elem_id="inp_block") as inp:
|
| 141 |
overlay = gr.Checkbox(label='覆盖原图', value=True)
|
| 142 |
inp_img = gr.ImageMask(label="上传图片", value=None, sources=["upload", "clipboard", "webcam"], interactive=True, type="pil", eraser=False, transforms=None, brush=gr.Brush(colors=['#FFFFFF'], color_mode='fixed'), layers=False, show_share_button=False, show_download_button=False)
|
| 143 |
-
inp_str = gr.Slider(label="重绘强度", value=1, minimum=0.01, maximum=1, step=0.01
|
| 144 |
reuse_img_inp = gr.Button(value='使用上一次生成的图片')
|
| 145 |
-
model.change(lambda m, i: gr.update(visible=m.startswith('nai-diffusion-4') and (i != None)), inputs=[model, inp_img], outputs=inp_str)
|
| 146 |
-
inp_img.change(lambda m, i: gr.update(visible=m.startswith('nai-diffusion-4') and (i != None)), inputs=[model, inp_img], outputs=inp_str)
|
| 147 |
selection = gr.Radio(choices=['i2i', 'inp'], value='i2i', visible=False)
|
| 148 |
with gr.Accordion("角色控制(仅nai4有效)", open=False, elem_id="v4chars_tab") as chars:
|
| 149 |
with gr.Row():
|
|
|
|
| 140 |
with gr.Tab('局部重绘', elem_id="inp_block") as inp:
|
| 141 |
overlay = gr.Checkbox(label='覆盖原图', value=True)
|
| 142 |
inp_img = gr.ImageMask(label="上传图片", value=None, sources=["upload", "clipboard", "webcam"], interactive=True, type="pil", eraser=False, transforms=None, brush=gr.Brush(colors=['#FFFFFF'], color_mode='fixed'), layers=False, show_share_button=False, show_download_button=False)
|
| 143 |
+
inp_str = gr.Slider(label="重绘强度", value=1, minimum=0.01, maximum=1, step=0.01)
|
| 144 |
reuse_img_inp = gr.Button(value='使用上一次生成的图片')
|
|
|
|
|
|
|
| 145 |
selection = gr.Radio(choices=['i2i', 'inp'], value='i2i', visible=False)
|
| 146 |
with gr.Accordion("角色控制(仅nai4有效)", open=False, elem_id="v4chars_tab") as chars:
|
| 147 |
with gr.Row():
|
utils.py
CHANGED
|
@@ -106,7 +106,6 @@ def generate_novelai_image(
|
|
| 106 |
"ucPreset": 0,
|
| 107 |
"cfg_rescale": cfg_rescale,
|
| 108 |
"controlnet_strength": 1,
|
| 109 |
-
"inpaintImg2ImgStrength": inp_str,
|
| 110 |
"dynamic_thresholding": dyn_threshold,
|
| 111 |
"params_version": 3,
|
| 112 |
"legacy": False,
|
|
@@ -152,6 +151,8 @@ def generate_novelai_image(
|
|
| 152 |
payload['parameters']['mask'] = image2base64(mask)
|
| 153 |
payload['parameters']['image'] = image2base64(inp_img['background'])
|
| 154 |
payload['parameters']['extra_noise_seed'] = seed
|
|
|
|
|
|
|
| 155 |
if i2i_image != None and selection == 'i2i':
|
| 156 |
payload['action'] = "img2img"
|
| 157 |
payload['parameters']['image'] = image2base64(i2i_image)
|
|
|
|
| 106 |
"ucPreset": 0,
|
| 107 |
"cfg_rescale": cfg_rescale,
|
| 108 |
"controlnet_strength": 1,
|
|
|
|
| 109 |
"dynamic_thresholding": dyn_threshold,
|
| 110 |
"params_version": 3,
|
| 111 |
"legacy": False,
|
|
|
|
| 151 |
payload['parameters']['mask'] = image2base64(mask)
|
| 152 |
payload['parameters']['image'] = image2base64(inp_img['background'])
|
| 153 |
payload['parameters']['extra_noise_seed'] = seed
|
| 154 |
+
payload['parameters']['inpaintImg2ImgStrength'] = inp_str
|
| 155 |
+
payload['parameters']['img2img'] = {'strength': inp_str, 'color_correct': True}
|
| 156 |
if i2i_image != None and selection == 'i2i':
|
| 157 |
payload['action'] = "img2img"
|
| 158 |
payload['parameters']['image'] = image2base64(i2i_image)
|