kadirnar commited on
Commit
14304cb
·
1 Parent(s): 1e1a292
diffusion_webui/controlnet/controlnet_hed.py CHANGED
@@ -91,8 +91,8 @@ def stable_diffusion_controlnet_hed_app():
91
  )
92
 
93
  controlnet_hed_model_id = gr.Dropdown(
94
- choices=stable_model_list,
95
- value=stable_model_list[1],
96
  label="ControlNet Model Id",
97
  )
98
 
 
91
  )
92
 
93
  controlnet_hed_model_id = gr.Dropdown(
94
+ choices=controlnet_hed_model_list,
95
+ value=controlnet_hed_model_list[1],
96
  label="ControlNet Model Id",
97
  )
98
 
diffusion_webui/controlnet/controlnet_pose.py CHANGED
@@ -92,8 +92,8 @@ def stable_diffusion_controlnet_pose_app():
92
  )
93
 
94
  controlnet_pose_model_id = gr.Dropdown(
95
- choices=stable_model_list,
96
- value=stable_model_list[1],
97
  label="ControlNet Model Id",
98
  )
99
 
 
92
  )
93
 
94
  controlnet_pose_model_id = gr.Dropdown(
95
+ choices=controlnet_pose_model_list,
96
+ value=controlnet_pose_model_list[1],
97
  label="ControlNet Model Id",
98
  )
99
 
diffusion_webui/controlnet/controlnet_scribble.py CHANGED
@@ -85,16 +85,16 @@ def stable_diffusion_controlnet_scribble_app():
85
  type="filepath", label="Image"
86
  )
87
 
88
- controlnet_scribble_stablev1_model_id = gr.Dropdown(
89
  choices=stable_model_list,
90
  value=stable_model_list[0],
91
  label="Stable v1.5 Model Id",
92
  )
93
 
94
- controlnet_scribble_stablev2_model_id = gr.Dropdown(
95
- choices=stable_model_list,
96
- value=stable_model_list[1],
97
- label="Stable v2.1 Model Id",
98
  )
99
 
100
  controlnet_scribble_prompt = gr.Textbox(
@@ -144,8 +144,8 @@ def stable_diffusion_controlnet_scribble_app():
144
  ],
145
  inputs=[
146
  controlnet_scribble_image_file,
147
- controlnet_scribble_stablev1_model_id,
148
- controlnet_scribble_stablev2_model_id,
149
  controlnet_scribble_prompt,
150
  controlnet_scribble_negative_prompt,
151
  controlnet_scribble_guidance_scale,
@@ -159,8 +159,8 @@ def stable_diffusion_controlnet_scribble_app():
159
  fn=stable_diffusion_controlnet_scribble,
160
  inputs=[
161
  controlnet_scribble_image_file,
162
- controlnet_scribble_stablev1_model_id,
163
- controlnet_scribble_stablev2_model_id,
164
  controlnet_scribble_prompt,
165
  controlnet_scribble_negative_prompt,
166
  controlnet_scribble_guidance_scale,
 
85
  type="filepath", label="Image"
86
  )
87
 
88
+ controlnet_scribble_stable_model_id = gr.Dropdown(
89
  choices=stable_model_list,
90
  value=stable_model_list[0],
91
  label="Stable v1.5 Model Id",
92
  )
93
 
94
+ controlnet_scribble_model_id = gr.Dropdown(
95
+ choices=controlnet_hed_model_list,
96
+ value=controlnet_hed_model_list[1],
97
+ label="ControlNet Model Id",
98
  )
99
 
100
  controlnet_scribble_prompt = gr.Textbox(
 
144
  ],
145
  inputs=[
146
  controlnet_scribble_image_file,
147
+ controlnet_scribble_stable_model_id,
148
+ controlnet_scribble_model_id,
149
  controlnet_scribble_prompt,
150
  controlnet_scribble_negative_prompt,
151
  controlnet_scribble_guidance_scale,
 
159
  fn=stable_diffusion_controlnet_scribble,
160
  inputs=[
161
  controlnet_scribble_image_file,
162
+ controlnet_scribble_stable_model_id,
163
+ controlnet_scribble_model_id,
164
  controlnet_scribble_prompt,
165
  controlnet_scribble_negative_prompt,
166
  controlnet_scribble_guidance_scale,