updated models.py to enable more apis

AIRI-AUBE · Apr 23, 2023 · c464f29 · c464f29
1 parent b5d78c1
commit c464f29
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/modules/api/api.py b/modules/api/api.py
@@ -807,9 +807,10 @@ def invocations(self, req: InvocationsRequest):
             elif req.task == 'get-upscalers':
                 response = self.get_upscalers()
                 return response
-            elif req.task == 'get-samplers':
-                response = self.get_samplers()
-                return response
+            # useless
+            # elif req.task == 'get-samplers':
+            #     response = self.get_samplers()
+            #     return response
             elif req.task == 'interrogate':
                 response = self.interrogateapi(req.interrogate_payload)
                 return response

diff --git a/modules/api/models.py b/modules/api/models.py
@@ -301,6 +301,9 @@ class InvocationsRequest(BaseModel):
     img2img_payload: Optional[StableDiffusionImg2ImgProcessingAPI]
     extras_single_payload: Optional[ExtrasSingleImageRequest]
     extras_batch_payload: Optional[ExtrasBatchImagesRequest]
+    progress_payload:Optional[ProgressRequest]
+    interrogate_payload:Optional[InterrogateRequest]
+    post_options_payload:Optional[]
 
 class InvocationsErrorResponse(BaseModel):
     error: str = Field(title="Invocation error", description="Error response from invocation.")