extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-06-20 01:48:21 +02:00
Code Issues Packages Projects Releases Wiki Activity

Restore VAE model in metadata files and remove prompt strength in txt2img generations

Browse Source
This commit is contained in:
Olivia Godone-Maresca 2023-02-06 23:07:23 -05:00
parent 5c6c2303ba
commit b3a988bc0b
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ui/easydiffusion/utils/save_utils.py
View File

@ -24,6 +24,7 @@ TASK_TEXT_MAPPING = {
'sampler_name': 'Sampler', 'sampler_name': 'Sampler',
'negative_prompt': 'Negative Prompt', 'negative_prompt': 'Negative Prompt',
'use_stable_diffusion_model': 'Stable Diffusion model', 'use_stable_diffusion_model': 'Stable Diffusion model',
'use_vae_model': 'VAE model',
'use_hypernetwork_model': 'Hypernetwork model', 'use_hypernetwork_model': 'Hypernetwork model',
'hypernetwork_strength': 'Hypernetwork Strength' 'hypernetwork_strength': 'Hypernetwork Strength'
} }
@ -71,6 +72,8 @@ def get_printable_request(req: GenerateImageRequest):
metadata = req.dict() metadata = req.dict()
del metadata['init_image'] del metadata['init_image']
del metadata['init_image_mask'] del metadata['init_image_mask']
if (req.init_image is None):
del metadata['prompt_strength']
return metadata return metadata
def make_filename_callback(req: GenerateImageRequest, suffix=None, now=None): def make_filename_callback(req: GenerateImageRequest, suffix=None, now=None):