mirror of
https://github.com/lllyasviel/stable-diffusion-webui-forge.git
synced 2026-01-31 13:29:46 +00:00
393c19bbcf
Forge use totally different mechanism to handle lora/lycoris models I think it is good to remove the back compact things so "lyco" alias can be used by legacy lycoris extensions. (some other extension may rely on it)
101 lines
3.8 KiB
Python
101 lines
3.8 KiB
Python
import re
|
|
|
|
import gradio as gr
|
|
from fastapi import FastAPI
|
|
|
|
import network
|
|
import networks
|
|
import lora # noqa:F401
|
|
import lora_patches
|
|
import extra_networks_lora
|
|
import ui_extra_networks_lora
|
|
from modules import script_callbacks, ui_extra_networks, extra_networks, shared
|
|
|
|
|
|
def unload():
|
|
networks.originals.undo()
|
|
|
|
|
|
def before_ui():
|
|
ui_extra_networks.register_page(ui_extra_networks_lora.ExtraNetworksPageLora())
|
|
|
|
networks.extra_network_lora = extra_networks_lora.ExtraNetworkLora()
|
|
extra_networks.register_extra_network(networks.extra_network_lora)
|
|
|
|
|
|
networks.originals = lora_patches.LoraPatches()
|
|
|
|
script_callbacks.on_model_loaded(networks.assign_network_names_to_compvis_modules)
|
|
script_callbacks.on_script_unloaded(unload)
|
|
script_callbacks.on_before_ui(before_ui)
|
|
script_callbacks.on_infotext_pasted(networks.infotext_pasted)
|
|
|
|
|
|
shared.options_templates.update(shared.options_section(('extra_networks', "Extra Networks"), {
|
|
"sd_lora": shared.OptionInfo("None", "Add network to prompt", gr.Dropdown, lambda: {"choices": ["None", *networks.available_networks]}, refresh=networks.list_available_networks),
|
|
"lora_preferred_name": shared.OptionInfo("Alias from file", "When adding to prompt, refer to Lora by", gr.Radio, {"choices": ["Alias from file", "Filename"]}),
|
|
"lora_add_hashes_to_infotext": shared.OptionInfo(True, "Add Lora hashes to infotext"),
|
|
"lora_show_all": shared.OptionInfo(False, "Always show all networks on the Lora page").info("otherwise, those detected as for incompatible version of Stable Diffusion will be hidden"),
|
|
"lora_hide_unknown_for_versions": shared.OptionInfo([], "Hide networks of unknown versions for model versions", gr.CheckboxGroup, {"choices": ["SD1", "SD2", "SDXL"]}),
|
|
"lora_in_memory_limit": shared.OptionInfo(0, "Number of Lora networks to keep cached in memory", gr.Number, {"precision": 0}),
|
|
"lora_not_found_warning_console": shared.OptionInfo(False, "Lora not found warning in console"),
|
|
"lora_not_found_gradio_warning": shared.OptionInfo(False, "Lora not found warning popup in webui"),
|
|
}))
|
|
|
|
|
|
shared.options_templates.update(shared.options_section(('compatibility', "Compatibility"), {
|
|
"lora_functional": shared.OptionInfo(False, "Lora/Networks: use old method that takes longer when you have multiple Loras active and produces same results as kohya-ss/sd-webui-additional-networks extension"),
|
|
}))
|
|
|
|
|
|
def create_lora_json(obj: network.NetworkOnDisk):
|
|
return {
|
|
"name": obj.name,
|
|
"alias": obj.alias,
|
|
"path": obj.filename,
|
|
"metadata": obj.metadata,
|
|
}
|
|
|
|
|
|
def api_networks(_: gr.Blocks, app: FastAPI):
|
|
@app.get("/sdapi/v1/loras")
|
|
async def get_loras():
|
|
return [create_lora_json(obj) for obj in networks.available_networks.values()]
|
|
|
|
@app.post("/sdapi/v1/refresh-loras")
|
|
async def refresh_loras():
|
|
return networks.list_available_networks()
|
|
|
|
|
|
script_callbacks.on_app_started(api_networks)
|
|
|
|
re_lora = re.compile("<lora:([^:]+):")
|
|
|
|
|
|
def infotext_pasted(infotext, d):
|
|
hashes = d.get("Lora hashes")
|
|
if not hashes:
|
|
return
|
|
|
|
hashes = [x.strip().split(':', 1) for x in hashes.split(",")]
|
|
hashes = {x[0].strip().replace(",", ""): x[1].strip() for x in hashes}
|
|
|
|
def network_replacement(m):
|
|
alias = m.group(1)
|
|
shorthash = hashes.get(alias)
|
|
if shorthash is None:
|
|
return m.group(0)
|
|
|
|
network_on_disk = networks.available_network_hash_lookup.get(shorthash)
|
|
if network_on_disk is None:
|
|
return m.group(0)
|
|
|
|
return f'<lora:{network_on_disk.get_alias()}:'
|
|
|
|
d["Prompt"] = re.sub(re_lora, network_replacement, d["Prompt"])
|
|
|
|
|
|
script_callbacks.on_infotext_pasted(infotext_pasted)
|
|
|
|
shared.opts.onchange("lora_in_memory_limit", networks.purge_networks_from_memory)
|