hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-04 11:12:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

lora extension rework to include other types of networks

Browse Source
This commit is contained in:
AUTOMATIC1111
2023-07-16 23:13:55 +03:00
parent 7d26c479ee
commit b75b004fe6
10 changed files with 777 additions and 589 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
extensions-builtin/Lora/ui_extra_networks_lora.py
View File

@@ -1,5 +1,5 @@
import os
import lora
import networks
from modules import shared, ui_extra_networks
from modules.ui_extra_networks import quote_js
@@ -11,10 +11,10 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage):
super().__init__('Lora')
def refresh(self):
lora.list_available_loras()
networks.list_available_networks()
def create_item(self, name, index=None):
lora_on_disk = lora.available_loras.get(name)
lora_on_disk = networks.available_networks.get(name)
path, ext = os.path.splitext(lora_on_disk.filename)
@@ -43,7 +43,7 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage):
return item
def list_items(self):
for index, name in enumerate(lora.available_loras):
for index, name in enumerate(networks.available_networks):
item = self.create_item(name, index)
yield item