diff --git a/decart/models.py b/decart/models.py index 00d986f..be35664 100644 --- a/decart/models.py +++ b/decart/models.py @@ -8,12 +8,15 @@ # Canonical names "lucy-2.1", "lucy-2.1-vton", + "lucy-vton-2", "lucy-restyle-2", # Latest aliases (server-side resolution) "lucy-latest", "lucy-vton-latest", "lucy-restyle-latest", # Deprecated names + "lucy-vton", + "lucy-2.1-vton-2", "mirage_v2", ] VideoModels = Literal[ @@ -21,13 +24,16 @@ "lucy-clip", "lucy-2.1", "lucy-2.1-vton", + "lucy-vton-2", "lucy-restyle-2", # Latest aliases (server-side resolution) "lucy-latest", "lucy-vton-latest", "lucy-restyle-latest", "lucy-clip-latest", - # Deprecated names + # Deprecated / alias names + "lucy-vton", + "lucy-2.1-vton-2", "lucy-pro-v2v", "lucy-restyle-v2v", ] @@ -47,6 +53,9 @@ # Video aliases "lucy-pro-v2v": "lucy-clip", "lucy-restyle-v2v": "lucy-restyle-2", + # VTON aliases + "lucy-vton": "lucy-2.1-vton", + "lucy-2.1-vton-2": "lucy-vton-2", # Image aliases "lucy-pro-i2i": "lucy-image-2", } @@ -207,6 +216,20 @@ class ImageToImageInput(DecartBaseModel): width=1088, height=624, ), + "lucy-vton-2": ModelDefinition( + name="lucy-vton-2", + url_path="/v1/stream", + fps=20, + width=1088, + height=624, + ), + "lucy-2.1-vton-2": ModelDefinition( + name="lucy-2.1-vton-2", + url_path="/v1/stream", + fps=20, + width=1088, + height=624, + ), "lucy-restyle-latest": ModelDefinition( name="lucy-restyle-latest", url_path="/v1/stream", @@ -215,6 +238,13 @@ class ImageToImageInput(DecartBaseModel): height=704, ), # Deprecated names + "lucy-vton": ModelDefinition( + name="lucy-vton", + url_path="/v1/stream", + fps=20, + width=1088, + height=624, + ), "mirage_v2": ModelDefinition( name="mirage_v2", url_path="/v1/stream", @@ -249,6 +279,30 @@ class ImageToImageInput(DecartBaseModel): height=624, input_schema=VideoEdit2Input, ), + "lucy-vton-2": ModelDefinition( + name="lucy-vton-2", + url_path="/v1/jobs/lucy-vton-2", + fps=20, + width=1088, + height=624, + input_schema=VideoEdit2Input, + ), + "lucy-2.1-vton-2": ModelDefinition( + name="lucy-2.1-vton-2", + url_path="/v1/jobs/lucy-2.1-vton-2", + fps=20, + width=1088, + height=624, + input_schema=VideoEdit2Input, + ), + "lucy-vton": ModelDefinition( + name="lucy-vton", + url_path="/v1/jobs/lucy-vton", + fps=20, + width=1088, + height=624, + input_schema=VideoEdit2Input, + ), "lucy-restyle-2": ModelDefinition( name="lucy-restyle-2", url_path="/v1/jobs/lucy-restyle-2",