[diffusion] api: add sampling parameters and model info endpoint to OpenAI API (#15071)

Co-authored-by: niehen6174 <niehen.6174@gmail.com>
Co-authored-by: Mick <mickjagger19@icloud.com>
Co-authored-by: niehen6174 <nihen6174@gmail.com>
This commit is contained in:
WenhaoZhang
2025-12-17 15:33:18 +08:00
committed by GitHub
parent 79ab57bd7a
commit cdce516331
5 changed files with 91 additions and 1 deletions

View File

@@ -29,6 +29,10 @@ class ImageGenerationsRequest(BaseModel):
seed: Optional[int] = 1024
generator_device: Optional[str] = "cuda"
user: Optional[str] = None
negative_prompt: Optional[str] = None
guidance_scale: Optional[float] = None
num_inference_steps: Optional[int] = None
enable_teacache: Optional[bool] = False
# Video API protocol models
@@ -62,6 +66,7 @@ class VideoGenerationsRequest(BaseModel):
guidance_scale: Optional[float] = None
guidance_scale_2: Optional[float] = None
negative_prompt: Optional[str] = None
enable_teacache: Optional[bool] = False
class VideoListResponse(BaseModel):