diff --git a/decart/models.py b/decart/models.py index 9d7aa58..b784df6 100644 --- a/decart/models.py +++ b/decart/models.py @@ -4,7 +4,9 @@ from .types import FileInput, MotionTrajectoryInput -RealTimeModels = Literal["mirage", "mirage_v2", "lucy_v2v_720p_rt", "avatar-live"] +RealTimeModels = Literal[ + "mirage", "mirage_v2", "lucy_v2v_720p_rt", "lucy_v2v_14b_rt", "avatar-live" +] VideoModels = Literal[ "lucy-dev-i2v", "lucy-fast-v2v", @@ -175,6 +177,14 @@ class ImageToImageInput(DecartBaseModel): height=704, input_schema=BaseModel, ), + "lucy_v2v_14b_rt": ModelDefinition( + name="lucy_v2v_14b_rt", + url_path="/v1/stream", + fps=15, + width=1280, + height=704, + input_schema=BaseModel, + ), "avatar-live": ModelDefinition( name="avatar-live", url_path="/v1/avatar-live/stream", diff --git a/tests/test_realtime_unit.py b/tests/test_realtime_unit.py index 15882ff..7f21b33 100644 --- a/tests/test_realtime_unit.py +++ b/tests/test_realtime_unit.py @@ -39,6 +39,17 @@ def test_realtime_models_available(): model2 = models.realtime("lucy_v2v_720p_rt") assert model2.name == "lucy_v2v_720p_rt" + assert model2.fps == 25 + assert model2.width == 1280 + assert model2.height == 704 + assert model2.url_path == "/v1/stream" + + model2 = models.realtime("lucy_v2v_14b_rt") + assert model2.name == "lucy_v2v_14b_rt" + assert model2.fps == 15 + assert model2.width == 1280 + assert model2.height == 704 + assert model2.url_path == "/v1/stream" @pytest.mark.asyncio