From 9aa82aa80a19474e410b3d09ed622483d6e9c82d Mon Sep 17 00:00:00 2001 From: Manuel Schmid Date: Fri, 2 Feb 2024 01:57:33 +0100 Subject: [PATCH] fix: add missing refiner in A1111 parse_json --- modules/meta_parser.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/modules/meta_parser.py b/modules/meta_parser.py index 1e12b6cd..a452686d 100644 --- a/modules/meta_parser.py +++ b/modules/meta_parser.py @@ -308,12 +308,13 @@ class A1111MetadataParser(MetadataParser): except Exception: pass - if 'base_model' in data: - for filename in modules.config.model_filenames: - path = Path(filename) - if data['base_model'] == path.stem: - data['base_model'] = filename - break + for key in ['base_model', 'refiner_model']: + if key in data: + for filename in modules.config.model_filenames: + path = Path(filename) + if data[key] == path.stem: + data[key] = filename + break if 'lora_hashes' in data: lora_filenames = modules.config.lora_filenames.copy()