Commit b1b9dcb1 authored by peiqi han's avatar peiqi han
Browse files

新增color ipadapter配置

parent d2e40433
No related merge requests found
Showing with 2 additions and 1 deletion
+2 -1
......@@ -783,6 +783,7 @@ class PlugableIPAdapter(torch.nn.Module):
weights_dict = {
'Full': [1]*layer_num,
'Ip_plus_LoRA': [0.58,0.62,0.62,0.7,0.61,0.58,0.6,0.64,0.57,0.6,0.58,0.59,0.59,0.63,0.59,0.46,0.28,0.38,0.36,0.21,0.5,0.47,0.51,0.62,0.48,0.4,0.0,0.04,0.16,0.3,0.25,0.48,0.58,0.51,0.58,0.55,0.42,0.42,0.47,0.52,0.53,0.56,0.61,0.6,0.66,0.67,0.64,0.55,0.57,0.51,0.61,0.6,0.63,0.63,0.63,0.7,0.66,0.66,0.57,0.63,0.61,0.54,0.51,0.6,0.57,0.55,0.6,0.6,0.6,0.63],
"Color_":[0.0]*34+[0.3]*10+[0.0]*10+[1.0]*6+[0.0]*10 #hpq_edit
}
IP_weights = weights_dict[IP_mode] if ip_layer_wise == '' \
else [float(i) for i in ip_layer_wise.split(',')]
......
......@@ -684,7 +684,7 @@ class ControlNetUiGroup(object):
self.multi_references_dir = gr.Textbox(label='Cache File Path', value='')
with gr.Column(scale=2):
self.IP_mode = gr.Dropdown(
['Full', 'Ip_plus_LoRA'],
['Full', 'Ip_plus_LoRA','Color_'],
label='IP Mode',
value='Full',
)
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment