23 lines
734 B
Python
23 lines
734 B
Python
from modules.api.models import *
|
|
from typing import Optional
|
|
|
|
|
|
class InvocationsRequest(BaseModel):
|
|
task: str
|
|
username: Optional[str]
|
|
# checkpoint_info:Optional[dict]
|
|
models: Optional[dict]
|
|
# txt2img_payload: Optional[StableDiffusionTxt2ImgProcessingAPI]
|
|
# img2img_payload: Optional[StableDiffusionImg2ImgProcessingAPI]
|
|
extras_single_payload: Optional[ExtrasSingleImageRequest]
|
|
extras_batch_payload: Optional[ExtrasBatchImagesRequest]
|
|
interrogate_payload: Optional[InterrogateRequest]
|
|
db_create_model_payload: Optional[str]
|
|
merge_checkpoint_payload: Optional[dict]
|
|
param_s3: Optional[str] = None
|
|
payload_string: Optional[str] = None
|
|
|
|
|
|
class PingResponse(BaseModel):
|
|
status: str
|