OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-05-06 20:09:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

dev

Browse Source
This commit is contained in:
Misha4ca228 2025-04-30 21:53:01 +03:00
parent 2a3be5ff79
commit 7c906a3ff0
1 changed files with 24 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
modules/api/api.py
View File

@ -577,6 +577,11 @@ class Api:
filename = str(uid_int)[0:18] filename = str(uid_int)[0:18]
folder = "ddcn_results" folder = "ddcn_results"
retry = 0
while retry <= 5:
try:
s3_client.put_object( s3_client.put_object(
Bucket=get_env_var(key="S3_BUCKET"), Bucket=get_env_var(key="S3_BUCKET"),
Key=f"{folder}/{filename}.jpg", Key=f"{folder}/{filename}.jpg",
@ -584,9 +589,20 @@ class Api:
ContentType="image/jpeg" ContentType="image/jpeg"
) )
s3_url = f"{get_env_var('S3_ENDPOINT')}/{get_env_var('S3_BUCKET')}/{folder}/{filename}.jpg" s3_url = f"{get_env_var('S3_ENDPOINT')}/{get_env_var('S3_BUCKET')}/{folder}/{filename}.jpg"
return models.ImageToImageResponse(parameters=vars(img2imgreq), info=processed.js(), s3_url=s3_url) return models.ImageToImageResponse(parameters=vars(img2imgreq), info=processed.js(), s3_url=s3_url)
except Exception as e:
print(f"Error save on s3: {e}")
retry =+ 1
continue
else:
raise HTTPException(status_code=500, detail="Error save on s3:")
def extras_single_image_api(self, req: models.ExtrasSingleImageRequest): def extras_single_image_api(self, req: models.ExtrasSingleImageRequest):
reqDict = setUpscalers(req) reqDict = setUpscalers(req)