Fix story maker bug

2023-09-22 00:09:21 +08:00
parent 4e45130140
commit 5c2dd13ab2
10 changed files with 147 additions and 133 deletions
@@ -118,7 +118,7 @@ class ComputeKernel:
                if task_req.state == ComputeTaskState.ERROR:
                    break

-                if check_times >= 20:
+                if check_times >= 120:
                    task_req.state = ComputeTaskState.ERROR
                    break

@@ -129,7 +129,7 @@ class ComputeKernel:
        if task_req.state == ComputeTaskState.DONE:
            return task_req.result

-        return "error!"
+        raise Exception("error!")

    
    def text_embedding(self,input:str,model_name:Optional[str] = None):