ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-04-22 01:20:46 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

upgraded novita ai video generator

Browse Source
This commit is contained in:
Saifeddine ALOUI 2025-03-07 22:23:27 +01:00
parent 888ea24534
commit d7f3a409f7
1 changed files with 5 additions and 1 deletions

6
lollms/services/ttv/novita_ai/lollms_novita_ai.py

@ -33,6 +33,7 @@ class LollmsNovitaAITextToVideo(LollmsTTV):
{"name":"api_key", "type":"str", "value":api_key, "help":"A valid Novita AI key to generate text using anthropic api"},
{"name":"generation_engine","type":"str","value":"stable_diffusion", "options": ["stable_diffusion", "hunyuan-video-fast", "wan-t2v"], "help":"The engine name"},
{"name":"sd_model_name","type":"str","value":"darkSushiMixMix_225D_64380.safetensors", "options": ["darkSushiMixMix_225D_64380.safetensors"], "help":"The model name"}
{"name":"n_frames","type":"int","value":129, "help":"The number of frames in the video"}
]),
BaseConfig(config={
"api_key": "", # use avx2
@ -73,7 +74,7 @@ class LollmsNovitaAITextToVideo(LollmsTTV):
width: int = 512,
steps: int = 20,
seed: int = -1,
nb_frames: int = 64,
nb_frames: int = None,
guidance_scale: Optional[float] = None,
loras: Optional[List[Dict[str, Any]]] = None,
embeddings: Optional[List[Dict[str, Any]]] = None,
@ -104,6 +105,9 @@ class LollmsNovitaAITextToVideo(LollmsTTV):
if output_dir is None:
output_dir = self.output_folder
if nb_frames is None:
nb_frames =self.service_config.n_frames
if self.service_config.generation_engine=="hunyuan-video-fast":
width, height, nb_frames, steps = self.pin_dimensions_frames_steps(width, height, nb_frames, steps)
url = "https://api.novita.ai/v3/async/hunyuan-video-fast"