diff --git a/modules/ui_extra_networks_lora.py b/modules/ui_extra_networks_lora.py index 53831d6bb..0538f96d9 100644 --- a/modules/ui_extra_networks_lora.py +++ b/modules/ui_extra_networks_lora.py @@ -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):