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

Configurable path to save to disk

Browse Source
This commit is contained in:
cmdr2
2022-09-05 17:25:25 +05:30
parent 7460e6f73b
commit 835dd4da9d
2 changed files with 33 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
ui/server.py
View File

@ -38,7 +38,7 @@ class ImageRequest(BaseModel):
seed: int = 42
prompt_strength: float = 0.8
# allow_nsfw: bool = False
save_to_disk: bool = False
save_to_disk_path: str = None
turbo: bool = True
use_cpu: bool = False
use_full_precision: bool = False
@ -90,9 +90,7 @@ async def image(req : ImageRequest):
r.turbo = req.turbo
r.use_cpu = req.use_cpu
r.use_full_precision = req.use_full_precision
if req.save_to_disk:
r.save_to_disk_path = outpath
r.save_to_disk_path = req.save_to_disk_path
try:
res: Response = runtime.mk_img(r)