mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-05-20 05:57:25 +08:00
b1f5a3254b
Adds a new `WavespeedTextToImageNode` that generates images via WaveSpeed's fast Flux inference API (flux-dev, flux-dev-fp8, flux-schnell, flux-schnell-fp8). Also adds the corresponding `WavespeedTextToImageRequest` Pydantic model to the wavespeed API module. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
from typing import Optional
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class WavespeedTextToImageRequest(BaseModel):
|
|
prompt: str = Field(...)
|
|
width: int = Field(1024)
|
|
height: int = Field(1024)
|
|
num_inference_steps: int = Field(28)
|
|
guidance_scale: float = Field(3.5)
|
|
seed: int = Field(-1)
|
|
enable_safety_checker: bool = Field(True)
|
|
enable_sync_mode: bool = Field(False)
|
|
num_images: int = Field(1)
|
|
output_format: str = Field("jpeg")
|
|
negative_prompt: Optional[str] = Field(None)
|
|
|
|
|
|
class SeedVR2ImageRequest(BaseModel):
|
|
image: str = Field(...)
|
|
target_resolution: str = Field(...)
|
|
output_format: str = Field("png")
|
|
enable_sync_mode: bool = Field(False)
|
|
|
|
|
|
class FlashVSRRequest(BaseModel):
|
|
target_resolution: str = Field(...)
|
|
video: str = Field(...)
|
|
duration: float = Field(...)
|
|
|
|
|
|
class TaskCreatedDataResponse(BaseModel):
|
|
id: str = Field(...)
|
|
|
|
|
|
class TaskCreatedResponse(BaseModel):
|
|
code: int = Field(...)
|
|
message: str = Field(...)
|
|
data: TaskCreatedDataResponse | None = Field(None)
|
|
|
|
|
|
class TaskResultDataResponse(BaseModel):
|
|
status: str = Field(...)
|
|
outputs: list[str] = Field([])
|
|
|
|
|
|
class TaskResultResponse(BaseModel):
|
|
code: int = Field(...)
|
|
message: str = Field(...)
|
|
data: TaskResultDataResponse | None = Field(None)
|