fix: add missing refiner in A1111 parse_json

This commit is contained in:
Manuel Schmid
2024-02-02 01:57:33 +01:00
parent f745d40687
commit 9aa82aa80a
+4 -3
View File
@@ -308,11 +308,12 @@ class A1111MetadataParser(MetadataParser):
except Exception: except Exception:
pass pass
if 'base_model' in data: for key in ['base_model', 'refiner_model']:
if key in data:
for filename in modules.config.model_filenames: for filename in modules.config.model_filenames:
path = Path(filename) path = Path(filename)
if data['base_model'] == path.stem: if data[key] == path.stem:
data['base_model'] = filename data[key] = filename
break break
if 'lora_hashes' in data: if 'lora_hashes' in data: