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

chore: specify too broad exception types

This commit is contained in:
Manuel Schmid 2024-02-04 01:31:24 +01:00
parent 8af73e622f
commit c668228fe8
No known key found for this signature in database
GPG Key ID: 32C4F7569B40B84B
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
modules/meta_parser.py
View File

@ -323,14 +323,16 @@ class A1111MetadataParser(MetadataParser):
if 'steps' in data and 'performance' not in data: if 'steps' in data and 'performance' not in data:
try: try:
data['performance'] = Performance[Steps(int(data['steps'])).name].value data['performance'] = Performance[Steps(int(data['steps'])).name].value
except Exception: except ValueError | KeyError:
pass pass
if 'sampler' in data: if 'sampler' in data:
sampler = data['sampler'].replace(' Karras', '') data['sampler'] = data['sampler'].replace(' Karras', '')
# get key # get key
data['sampler'] = [k for k, v in SAMPLERS.items() if v == sampler][0] for k, v in SAMPLERS.items():
if v == data['sampler']:
data['sampler'] = k
break
for key in ['base_model', 'refiner_model']: for key in ['base_model', 'refiner_model']:
if key in data: if key in data:
@ -497,7 +499,7 @@ def read_info_from_image(filepath) -> tuple[str | None, dict, MetadataScheme | N
try: try:
metadata_scheme = MetadataScheme(items.pop('fooocus_scheme', None)) metadata_scheme = MetadataScheme(items.pop('fooocus_scheme', None))
except Exception: except ValueError:
metadata_scheme = None metadata_scheme = None
# broad fallback # broad fallback