Moved some stuff

This commit is contained in:
ManInDark 2023-08-04 23:22:24 +02:00
parent 7a3314a927
commit d48f566389
No known key found for this signature in database
GPG Key ID: 72EC12A5B2D62779

View File

@ -159,8 +159,7 @@ def save_images_to_disk(
from easydiffusion.easydb.database import SessionLocal from easydiffusion.easydb.database import SessionLocal
session = SessionLocal() session = SessionLocal()
session.add(Image(
img = Image(
path = path_i, path = path_i,
seed = metadata_entries[i]["seed"], seed = metadata_entries[i]["seed"],
use_stable_diffusion_model = metadata_entries[i]["use_stable_diffusion_model"], use_stable_diffusion_model = metadata_entries[i]["use_stable_diffusion_model"],
@ -176,9 +175,7 @@ def save_images_to_disk(
tiling = metadata_entries[i]["tiling"], tiling = metadata_entries[i]["tiling"],
use_face_correction = metadata_entries[i]["use_face_correction"], use_face_correction = metadata_entries[i]["use_face_correction"],
use_upscale = metadata_entries[i]["use_upscale"] use_upscale = metadata_entries[i]["use_upscale"]
) ))
session.add(img)
session.commit() session.commit()
session.close() session.close()