File size: 2,383 Bytes
f50a305 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 | def inject(assembler, chain_definition, chain_items):
if not chain_items:
return
ksampler_name = chain_definition.get('ksampler_node', 'ksampler')
if ksampler_name not in assembler.node_map:
print(f"Warning: KSampler node '{ksampler_name}' not found for Flux1 IPAdapter chain. Skipping.")
return
ksampler_id = assembler.node_map[ksampler_name]
if 'model' not in assembler.workflow[ksampler_id]['inputs']:
print(f"Warning: KSampler node '{ksampler_name}' is missing 'model' input. Skipping Flux1 IPAdapter chain.")
return
current_model_connection = assembler.workflow[ksampler_id]['inputs']['model']
for item_data in chain_items:
image_loader_id = assembler._get_unique_id()
image_loader_node = assembler._get_node_template("LoadImage")
image_loader_node['inputs']['image'] = item_data['image']
assembler.workflow[image_loader_id] = image_loader_node
ipadapter_loader_id = assembler._get_unique_id()
ipadapter_loader_node = assembler._get_node_template("IPAdapterFluxLoader")
ipadapter_loader_node['inputs']['ipadapter'] = "ip-adapter.bin"
ipadapter_loader_node['inputs']['clip_vision'] = "google/siglip-so400m-patch14-384"
ipadapter_loader_node['inputs']['provider'] = "cuda"
assembler.workflow[ipadapter_loader_id] = ipadapter_loader_node
apply_ipa_id = assembler._get_unique_id()
apply_ipa_node = assembler._get_node_template("ApplyIPAdapterFlux")
apply_ipa_node['inputs']['weight'] = item_data['weight']
apply_ipa_node['inputs']['start_percent'] = item_data.get('start_percent', 0.0)
apply_ipa_node['inputs']['end_percent'] = item_data.get('end_percent', 0.6)
apply_ipa_node['inputs']['model'] = current_model_connection
apply_ipa_node['inputs']['ipadapter_flux'] = [ipadapter_loader_id, 0]
apply_ipa_node['inputs']['image'] = [image_loader_id, 0]
assembler.workflow[apply_ipa_id] = apply_ipa_node
current_model_connection = [apply_ipa_id, 0]
assembler.workflow[ksampler_id]['inputs']['model'] = current_model_connection
print(f"Flux1 IPAdapter injector applied. KSampler model input re-routed through {len(chain_items)} IPAdapter(s).") |