[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:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user