gyt/opendan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Return result code from sd node

Browse Source
This commit is contained in:
Song
2023-09-27 19:37:13 +08:00
committed by tsukasa
parent 5b6de8b82b
commit 2938a54402
6 changed files with 70 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/aios_kernel/compute_kernel.py
+10 -8
View File
@@ -175,17 +175,19 @@ class ComputeKernel:
return task_result.result
def text_2_image(self, prompt:str, model_name:Optional[str] = None):
def text_2_image(self, prompt:str, model_name:Optional[str] = None, negative_prompt = None):
task = ComputeTask()
task.set_text_2_image_params(prompt,model_name)
task.set_text_2_image_params(prompt,model_name, negative_prompt)
self.run(task)
return task
async def do_text_2_image(self, prompt:str, model_name:Optional[str] = None) -> [str, ComputeTaskResult]:
task_req = self.text_2_image(prompt,model_name)
task_result = self._send_task(task_req)
if task_req.state == ComputeTaskState.DONE:
return None, task_result
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._send_task(task)
return task_req.error_str, None
return task.result
# if task_req.state == ComputeTaskState.DONE:
# return None, task_result
# return task_req.error_str, None