chenchenghao/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[diffusion] feat: Add Configurable Generator Device and Seed Support via API (#14366)

Browse Source 
Co-authored-by: niehen6174 <niehen.6174@gmail.com>
Co-authored-by: Mick <mickjagger19@icloud.com>
This commit is contained in:
WenhaoZhang
2025-12-05 00:25:09 +08:00
committed by GitHub
co-authored by niehen6174 Mick
parent 11d33c0e8f
commit 788628b56f
8 changed files with 43 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/multimodal_gen/runtime/entrypoints/openai/protocol.py
+4
View File
@@ -26,6 +26,8 @@ class ImageGenerationsRequest(BaseModel):
style: Optional[str] = "vivid"
background: Optional[str] = "auto" # transparent | opaque | auto
output_format: Optional[str] = None # png | jpeg | webp
seed: Optional[int] = 1024
generator_device: Optional[str] = "cuda"
user: Optional[str] = None
@@ -54,6 +56,8 @@ class VideoGenerationsRequest(BaseModel):
size: Optional[str] = "720x1280"
fps: Optional[int] = None
num_frames: Optional[int] = None
seed: Optional[int] = 1024
generator_device: Optional[str] = "cuda"
class VideoListResponse(BaseModel):