diff --git a/decart/models.py b/decart/models.py index f15e084..630d1e2 100644 --- a/decart/models.py +++ b/decart/models.py @@ -4,9 +4,7 @@ from .types import FileInput, MotionTrajectoryInput -RealTimeModels = Literal[ - "mirage", "mirage_v2", "lucy_v2v_720p_rt", "lucy_v2v_14b_rt", "avatar-live" -] +RealTimeModels = Literal["mirage", "mirage_v2", "lucy_v2v_720p_rt", "lucy_2_rt", "avatar-live"] VideoModels = Literal[ "lucy-dev-i2v", "lucy-fast-v2v", @@ -177,8 +175,8 @@ class ImageToImageInput(DecartBaseModel): height=704, input_schema=BaseModel, ), - "lucy_v2v_14b_rt": ModelDefinition( - name="lucy_v2v_14b_rt", + "lucy_2_rt": ModelDefinition( + name="lucy_2_rt", url_path="/v1/stream", fps=20, width=1280, diff --git a/tests/test_realtime_unit.py b/tests/test_realtime_unit.py index f07279d..511b1b1 100644 --- a/tests/test_realtime_unit.py +++ b/tests/test_realtime_unit.py @@ -44,8 +44,8 @@ def test_realtime_models_available(): assert model2.height == 704 assert model2.url_path == "/v1/stream" - model2 = models.realtime("lucy_v2v_14b_rt") - assert model2.name == "lucy_v2v_14b_rt" + model2 = models.realtime("lucy_2_rt") + assert model2.name == "lucy_2_rt" assert model2.fps == 20 assert model2.width == 1280 assert model2.height == 720 diff --git a/uv.lock b/uv.lock index 5edbd6f..96ac2ab 100644 --- a/uv.lock +++ b/uv.lock @@ -597,7 +597,7 @@ wheels = [ [[package]] name = "decart" -version = "0.0.16" +version = "0.0.17" source = { editable = "." } dependencies = [ { name = "aiofiles" },