P01yH3dr0n commited on
Commit
40d7c7f
·
1 Parent(s): 30b02ae
Files changed (2) hide show
  1. app.py +1 -3
  2. utils.py +2 -1
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, visible=False)
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)