DominikDoom/a1111-sd-webui-tagcomplete
1
0
Fork 0
mirror of https://github.com/DominikDoom/a1111-sd-webui-tagcomplete.git synced 2026-01-26 11:09:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix lora import in a1111

Browse Source 
This makes the built-in list method work on the initial load there
This commit is contained in:
DominikDoom
2023-12-12 13:46:51 +01:00
parent 41626d22c3
commit 886704e351
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
scripts/tag_autocomplete_helper.py
View File

@@ -246,20 +246,22 @@ def _get_lyco():
# Attempt to use the build-in Lora.networks Lora/LyCORIS models lists.
try:
import importlib
lora_networks = importlib.import_module("extensions-builtin.Lora.networks")
import sys
from modules import extensions
sys.path.append(Path(extensions.extensions_builtin_dir).joinpath("Lora").as_posix())
import lora # pyright: ignore [reportMissingImports]
def _get_lora():
return [
Path(model.filename).absolute()
for model in lora_networks.available_networks.values()
for model in lora.available_loras.values()
if Path(model.filename).absolute().is_relative_to(LORA_PATH)
]
def _get_lyco():
return [
Path(model.filename).absolute()
for model in lora_networks.available_networks.values()
for model in lora.available_loras.values()
if Path(model.filename).absolute().is_relative_to(LYCO_PATH)
]