Archives/imaginAIry
2
0
Fork 0
mirror of https://github.com/brycedrennan/imaginAIry synced 2024-11-05 12:00:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

feature: have initial generated images in their own folder

Browse Source
This commit is contained in:
Bryce 2022-09-24 10:24:51 -07:00 committed by Bryce Drennan
parent 30a80faf47
commit 2eec756fab
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
imaginairy/api.py
View File

@ -124,8 +124,8 @@ def imagine_image_files(
): ):
prompt = result.prompt prompt = result.prompt
basefilename = f"{base_count:06}_{prompt.seed}_{prompt.sampler_type}{prompt.steps}_PS{prompt.prompt_strength}_{prompt_normalized(prompt.prompt_text)}" basefilename = f"{base_count:06}_{prompt.seed}_{prompt.sampler_type}{prompt.steps}_PS{prompt.prompt_strength}_{prompt_normalized(prompt.prompt_text)}"
filepath = os.path.join(outdir, f"{basefilename}.jpg") filepath = os.path.join(outdir, "generated", f"{basefilename}.jpg")
os.makedirs(filepath, exist_ok=True)
result.save(filepath) result.save(filepath)
logger.info(f" 🖼 saved to: {filepath}") logger.info(f" 🖼 saved to: {filepath}")
if result.upscaled_img: if result.upscaled_img: