Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

using string.format for filename formatting

Browse Source
This commit is contained in:
plucked 2022-10-07 09:24:39 +00:00
parent 89c6c11214
commit 395445e7b0
2 changed files with 14 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ldm/dream/args.py
View File

@ -562,9 +562,9 @@ class Args(object):
)
render_group.add_argument(
'--fnformat',
default='%(prefix)s.%(seed)s.png',
default='{prefix}.{seed}.png',
type=str,
help='Specify the template of the generated files names.',
help='Overwrite the filename format. You can use any argument as wildcard enclosed in curly braces. Default is {prefix}.{seed}.png',
)
render_group.add_argument(
'--grid',

30
scripts/dream.py
View File

@ -356,7 +356,7 @@ def main_loop(gen, opt, infile):
grid_img = make_grid(list(grid_images.values()))
grid_seeds = list(grid_images.keys())
first_seed = last_results[0][1]
filename = f'{prefix}.{first_seed}.png'
filename = f'{prefix}.png'
formatted_dream_prompt = opt.dream_prompt_str(seed=first_seed,grid=True,iterations=len(grid_images))
formatted_dream_prompt += f' # {grid_seeds}'
metadata = metadata_dumps(
@ -458,23 +458,17 @@ def prepare_image_metadata(
if postprocessed and opt.save_original:
filename = choose_postprocess_name(opt,prefix,seed)
else:
filename = opt.fnformat % {
'prefix': prefix,
'seed': seed,
'steps': opt.steps,
'prompt': opt.prompt,
'width': opt.width,
'height': opt.height,
'cfg_scale': opt.cfg_scale,
'perlin': opt.perlin,
'threshold': opt.threshold,
'gfpgan_strength': opt.gfpgan_strength,
'outcrop': opt.outcrop,
'upscale': opt.upscale,
'embiggen': opt.embiggen,
'embiggen_tiles': opt.embiggen_tiles,
'out_direction': opt.out_direction
}
wildcards = dict(opt.__dict__)
wildcards['prefix'] = prefix
wildcards['seed'] = seed
try:
filename = opt.fnformat.format(**wildcards)
except KeyError as e:
print(f'The filename format contains an unknown key \'{e.args[0]}\'. Will use \'{{prefix}}.{{seed}}.png\' instead')
filename = f'{prefix}.{seed}.png'
except IndexError as e:
print(f'The filename format is broken or complete. Will use \'{{prefix}}.{{seed}}.png\' instead')
filename = f'{prefix}.{seed}.png'
if opt.variation_amount > 0:
first_seed = first_seed or seed