66
77RealTimeModels = Literal ["mirage" , "mirage_v2" , "lucy_v2v_720p_rt" , "lucy_2_rt" , "live_avatar" ]
88VideoModels = Literal [
9- "lucy-dev-i2v" ,
10- "lucy-fast-v2v" ,
11- "lucy-pro-t2v" ,
12- "lucy-pro-i2v" ,
139 "lucy-pro-v2v" ,
1410 "lucy-motion" ,
1511 "lucy-restyle-v2v" ,
1612 "lucy-2-v2v" ,
1713]
18- ImageModels = Literal ["lucy-pro-t2i" , "lucy-pro- i2i" ]
14+ ImageModels = Literal ["lucy-pro-i2i" ]
1915Model = Literal [RealTimeModels , VideoModels , ImageModels ]
2016
2117# Type variable for model name
@@ -46,24 +42,6 @@ class ModelDefinition(DecartBaseModel, Generic[ModelT]):
4642"""Type alias for model definitions that support realtime streaming."""
4743
4844
49- class TextToVideoInput (BaseModel ):
50- prompt : str = Field (..., min_length = 1 , max_length = 1000 )
51- seed : Optional [int ] = None
52- resolution : Optional [str ] = None
53- orientation : Optional [str ] = None
54-
55-
56- class ImageToVideoInput (DecartBaseModel ):
57- prompt : str = Field (
58- ...,
59- min_length = 1 ,
60- max_length = 1000 ,
61- )
62- data : FileInput
63- seed : Optional [int ] = None
64- resolution : Optional [str ] = None
65-
66-
6745class VideoToVideoInput (DecartBaseModel ):
6846 prompt : str = Field (
6947 ...,
@@ -128,17 +106,6 @@ class VideoEdit2Input(DecartBaseModel):
128106 enhance_prompt : Optional [bool ] = None
129107
130108
131- class TextToImageInput (BaseModel ):
132- prompt : str = Field (
133- ...,
134- min_length = 1 ,
135- max_length = 1000 ,
136- )
137- seed : Optional [int ] = None
138- resolution : Optional [str ] = None
139- orientation : Optional [str ] = None
140-
141-
142109class ImageToImageInput (DecartBaseModel ):
143110 prompt : str = Field (
144111 ...,
@@ -195,38 +162,6 @@ class ImageToImageInput(DecartBaseModel):
195162 ),
196163 },
197164 "video" : {
198- "lucy-dev-i2v" : ModelDefinition (
199- name = "lucy-dev-i2v" ,
200- url_path = "/v1/generate/lucy-dev-i2v" ,
201- fps = 25 ,
202- width = 1280 ,
203- height = 704 ,
204- input_schema = ImageToVideoInput ,
205- ),
206- "lucy-fast-v2v" : ModelDefinition (
207- name = "lucy-fast-v2v" ,
208- url_path = "/v1/generate/lucy-fast-v2v" ,
209- fps = 25 ,
210- width = 1280 ,
211- height = 704 ,
212- input_schema = VideoToVideoInput ,
213- ),
214- "lucy-pro-t2v" : ModelDefinition (
215- name = "lucy-pro-t2v" ,
216- url_path = "/v1/generate/lucy-pro-t2v" ,
217- fps = 25 ,
218- width = 1280 ,
219- height = 704 ,
220- input_schema = TextToVideoInput ,
221- ),
222- "lucy-pro-i2v" : ModelDefinition (
223- name = "lucy-pro-i2v" ,
224- url_path = "/v1/generate/lucy-pro-i2v" ,
225- fps = 25 ,
226- width = 1280 ,
227- height = 704 ,
228- input_schema = ImageToVideoInput ,
229- ),
230165 "lucy-pro-v2v" : ModelDefinition (
231166 name = "lucy-pro-v2v" ,
232167 url_path = "/v1/generate/lucy-pro-v2v" ,
@@ -261,14 +196,6 @@ class ImageToImageInput(DecartBaseModel):
261196 ),
262197 },
263198 "image" : {
264- "lucy-pro-t2i" : ModelDefinition (
265- name = "lucy-pro-t2i" ,
266- url_path = "/v1/generate/lucy-pro-t2i" ,
267- fps = 25 ,
268- width = 1280 ,
269- height = 704 ,
270- input_schema = TextToImageInput ,
271- ),
272199 "lucy-pro-i2i" : ModelDefinition (
273200 name = "lucy-pro-i2i" ,
274201 url_path = "/v1/generate/lucy-pro-i2i" ,
@@ -297,11 +224,7 @@ def video(model: VideoModels) -> VideoModelDefinition:
297224 Video models only support the queue API.
298225
299226 Available models:
300- - "lucy-pro-t2v" - Text-to-video
301- - "lucy-pro-i2v" - Image-to-video
302227 - "lucy-pro-v2v" - Video-to-video
303- - "lucy-dev-i2v" - Image-to-video (Dev quality)
304- - "lucy-fast-v2v" - Video-to-video (Fast quality)
305228 - "lucy-motion" - Image-to-motion-video
306229 - "lucy-restyle-v2v" - Video-to-video with prompt or reference image
307230 - "lucy-2-v2v" - Video-to-video editing (long-form, 720p)
@@ -318,7 +241,6 @@ def image(model: ImageModels) -> ImageModelDefinition:
318241 Image models only support the process (sync) API.
319242
320243 Available models:
321- - "lucy-pro-t2i" - Text-to-image
322244 - "lucy-pro-i2i" - Image-to-image
323245 """
324246 try :
0 commit comments