Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat: add enums for Performance, Steps and StepsUOV

Browse Source 
also move MetadataSchema enum to prevent circular dependency
This commit is contained in:
Manuel Schmid
2024-01-28 20:01:33 +01:00
parent 7ddd4e5209
commit cbc63ebba3
6 changed files with 93 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/private_logger.py
+2 -4
View File
@@ -20,9 +20,7 @@ def get_current_html_path():
return html_name
def log(img, metadata, save_metadata_to_image=False, metadata_scheme: str = MetadataScheme.FOOOCUS.value):
assert metadata_scheme in [item.value for item in MetadataScheme]
def log(img, metadata, save_metadata_to_image=False, metadata_scheme: MetadataScheme = MetadataScheme.FOOOCUS):
if args_manager.args.disable_image_log:
return
@@ -35,7 +33,7 @@ def log(img, metadata, save_metadata_to_image=False, metadata_scheme: str = Meta
pnginfo = PngInfo()
pnginfo.add_text('parameters', parsed_parameters)
pnginfo.add_text('fooocus_scheme', metadata_scheme)
pnginfo.add_text('fooocus_scheme', metadata_scheme.value)
else:
pnginfo = None
Image.fromarray(img).save(local_temp_filename, pnginfo=pnginfo)