S3保存路径修复

Lassi-Ki · Feb 28, 2024 · 8a47362 · 8a47362
1 parent 0a78e12
commit 8a47362
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/extensions/sd_EasyPhoto/scripts/easyphoto_infer.py b/extensions/sd_EasyPhoto/scripts/easyphoto_infer.py
@@ -1766,7 +1766,7 @@ def easyphoto_infer_forward(
                 output_image = template_image
 
             # Save the output image
-            output_path = f'./outputs_easyphoto/{user_ids[0]}/'
+            output_path = f'./outputs_easyphoto/{user_id}/{user_ids[0]}/'
             if output_path is not None:
                 # 如果文件夹不存在就创建它
                 if not os.path.exists(output_path):

diff --git a/modules/shared.py b/modules/shared.py
@@ -166,7 +166,7 @@ def upload_image(output_file_path, user_id, unique_id):
         if key.endswith('/'):
             key = key[:-1]
         key += "/" + user_id
-        __s3file = f'{key}/{unique_id}_{file_name}'
+        __s3file = f'{key}/{unique_id}/{file_name}'
         print(output_file_path, __s3file)
         s3_client.upload_file(output_file_path, bucket, __s3file)
     except ClientError as e: