happyz
/
Fooocus
mirror of https://github.com/lllyasviel/Fooocus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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
No known key found for this signature in database
GPG Key ID: 32C4F7569B40B84B
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
modules/meta_parser.py
View File

@ -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()