From 704d5300dee1b4e42107f48dd26f4d470856e408 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Enes=20Sad=C4=B1k=20=C3=96zbek?= Date: Tue, 16 Jan 2024 23:23:34 +0300 Subject: [PATCH] Add FaceID mode API parameter --- modules/api/api.py | 1 + modules/api/models.py | 3 ++- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/modules/api/api.py b/modules/api/api.py index 1a0eeab56..b6f5db96c 100644 --- a/modules/api/api.py +++ b/modules/api/api.py @@ -264,6 +264,7 @@ class Api: if hasattr(request, "face_id") and request.face_id and not request.script_name and (not request.alwayson_scripts or "FaceID" not in request.alwayson_scripts.keys()): request.script_name = "FaceID" request.script_args = [ + request.face_id.mode, request.face_id.model, request.face_id.scale, request.face_id.image, diff --git a/modules/api/models.py b/modules/api/models.py index 86b2997af..8ea5eb8f1 100644 --- a/modules/api/models.py +++ b/modules/api/models.py @@ -99,7 +99,8 @@ class IPAdapterItem(BaseModel): class FaceIDItem(BaseModel): - model: str = Field(title="Model", default="FaceID Base",description="The FaceID model to use.") + mode: list[str] = Field(title="Mode", default=["FaceID"], description="The mode to use (available values: FaceID, FaceSwap).") + model: str = Field(title="Model", default="FaceID Base", description="The FaceID model to use.") image: str = Field(title="Image", default="", description="Source face image, must be a base64 string containing the image's data.") scale: float = Field(title="Scale", default=1, gt=0, le=1, description="Scale of the source face, must be between 0 and 1.") structure: float = Field(title="Structure", default=1, gt=0, le=1, description="Structure to use, must be between 0 and 1.")