1580158
/
automatic
mirror of https://github.com/vladmandic/automatic
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

merge: modules/ui_extra_networks_lora.py

pull/4678/head
vladmandic 2026-03-12 14:16:51 +01:00
parent 6f13f6131e
commit 886341b1cc
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
modules/ui_extra_networks_lora.py
View File

@ -65,9 +65,17 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage):
clean_tags.pop('dataset', None)
return clean_tags
_VERSION_DISPLAY = {
'f1': 'Flux', 'sd1': 'SD 1.5', 'sd2': 'SD 2', 'xl': 'SDXL',
'sd3': 'SD3', 'sc': 'Cascade', 'hv': 'HunyuanVideo',
'chroma': 'Chroma', 'zimage': 'zImage', 'qwen': 'Qwen',
}
def cleanup_version(self, dct, lora):
ver = dct.get("baseModel", lora.sd_version)
ver = ver.replace(' 0.9', '').replace(' 1.0', '').replace(' ', '')
ver = self._VERSION_DISPLAY.get(ver, ver)
for suffix in (' 0.9', ' 1.0'): # strip uninformative minor versions
ver = ver.replace(suffix, '')
return ver
def create_item(self, name):