Fix the dall e invocation process

2023-12-04 10:39:56 +08:00
parent 83149b5590
commit 3dc4fe8e4c
7 changed files with 20 additions and 9 deletions
@@ -240,9 +240,9 @@ class ComputeKernel:

    async def do_text_2_image(self, prompt:str, model_name:Optional[str] = None, negative_prompt = None) -> ComputeTaskResult:
        task = self.text_2_image(prompt,model_name, negative_prompt)
-        task = await self._wait_task(task)
+        task_result = await self._wait_task(task)

-        return task.result
+        return task_result
        # if task_req.state == ComputeTaskState.DONE:
        #     return None, task_result