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
+7 -6
View File
@@ -308,12 +308,13 @@ class A1111MetadataParser(MetadataParser):
except Exception: except Exception:
pass pass
if 'base_model' in data: for key in ['base_model', 'refiner_model']:
for filename in modules.config.model_filenames: if key in data:
path = Path(filename) for filename in modules.config.model_filenames:
if data['base_model'] == path.stem: path = Path(filename)
data['base_model'] = filename if data[key] == path.stem:
break data[key] = filename
break
if 'lora_hashes' in data: if 'lora_hashes' in data:
lora_filenames = modules.config.lora_filenames.copy() lora_filenames = modules.config.lora_filenames.copy()