浏览代码

fix: async image gen automatic1111

Timothy J. Baek 8 月之前
父节点
当前提交
e442b3b169
共有 1 个文件被更改,包括 4 次插入2 次删除
  1. 4 2
      backend/apps/images/main.py

+ 4 - 2
backend/apps/images/main.py

@@ -15,6 +15,7 @@ import json
 import logging
 import re
 import requests
+import asyncio
 
 from utils.utils import (
     get_verified_user,
@@ -533,7 +534,9 @@ async def image_generations(
             if form_data.negative_prompt is not None:
                 data["negative_prompt"] = form_data.negative_prompt
 
-            r = requests.post(
+            # Use asyncio.to_thread for the requests.post call
+            r = await asyncio.to_thread(
+                requests.post,
                 url=f"{app.state.config.AUTOMATIC1111_BASE_URL}/sdapi/v1/txt2img",
                 json=data,
                 headers={"authorization": get_automatic1111_api_auth()},
@@ -553,7 +556,6 @@ async def image_generations(
                     json.dump({**data, "info": res["info"]}, f)
 
             return images
-
     except Exception as e:
         error = e
         if r != None: