[diffusion] feat: add support for additional sampling parameters in video generation API (#15062)

Co-authored-by: Mick <mickjagger19@icloud.com>
2025-12-14 19:44:03 +08:00
parent fdfabb7afc
commit e3f51e823e
3 changed files with 35 additions and 12 deletions
@@ -58,6 +58,10 @@ class VideoGenerationsRequest(BaseModel):
    num_frames: Optional[int] = None
    seed: Optional[int] = 1024
    generator_device: Optional[str] = "cuda"
+    num_inference_steps: Optional[int] = None
+    guidance_scale: Optional[float] = None
+    guidance_scale_2: Optional[float] = None
+    negative_prompt: Optional[str] = None


 class VideoListResponse(BaseModel):