mirror of
https://github.com/vladmandic/automatic
synced 2026-09-20 01:31:13 +02:00
merge: modules/ui_extra_networks_lora.py
This commit is contained in:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user