diff --git a/decart/models.py b/decart/models.py index 97aa076..a77095d 100644 --- a/decart/models.py +++ b/decart/models.py @@ -11,6 +11,7 @@ "lucy-vton-2", "lucy-vton-3", "lucy-vton-3.5", + "lucy-vton-3.5.2", "lucy-restyle-2", # Latest aliases (server-side resolution) "lucy-latest", @@ -27,6 +28,7 @@ "lucy-vton-2", "lucy-vton-3", "lucy-vton-3.5", + "lucy-vton-3.5.2", "lucy-restyle-2", # Latest aliases (server-side resolution) "lucy-latest", @@ -236,6 +238,13 @@ class ImageToImageInput(DecartBaseModel): width=1280, height=720, ), + "lucy-vton-3.5.2": ModelDefinition( + name="lucy-vton-3.5.2", + url_path="/v1/stream", + fps=30, + width=1280, + height=720, + ), "lucy-2.1-vton-2": ModelDefinition( name="lucy-2.1-vton-2", url_path="/v1/stream", @@ -301,6 +310,14 @@ class ImageToImageInput(DecartBaseModel): height=720, input_schema=VideoEdit2Input, ), + "lucy-vton-3.5.2": ModelDefinition( + name="lucy-vton-3.5.2", + url_path="/v1/jobs/lucy-vton-3.5.2", + fps=20, + width=1280, + height=720, + input_schema=VideoEdit2Input, + ), "lucy-2.1-vton-2": ModelDefinition( name="lucy-2.1-vton-2", url_path="/v1/jobs/lucy-2.1-vton-2", diff --git a/tests/test_models.py b/tests/test_models.py index 1f02afb..aa85137 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -46,6 +46,13 @@ def test_canonical_realtime_models() -> None: assert model.width == 1280 assert model.height == 720 + model = models.realtime("lucy-vton-3.5.2") + assert model.name == "lucy-vton-3.5.2" + assert model.url_path == "/v1/stream" + assert model.fps == 30 + assert model.width == 1280 + assert model.height == 720 + def test_deprecated_realtime_models() -> None: _warned_aliases.clear() @@ -99,6 +106,13 @@ def test_canonical_video_models() -> None: assert model.width == 1280 assert model.height == 720 + model = models.video("lucy-vton-3.5.2") + assert model.name == "lucy-vton-3.5.2" + assert model.url_path == "/v1/jobs/lucy-vton-3.5.2" + assert model.fps == 20 + assert model.width == 1280 + assert model.height == 720 + model = models.video("lucy-restyle-2") assert model.name == "lucy-restyle-2" assert model.url_path == "/v1/jobs/lucy-restyle-2"