razv23andClaude
728d83611f
Moved all files from comfyui-razv-wavespeed-custom/ subfolder directly to razv-wavespeed/ root directory to fix ComfyUI custom node loading. ComfyUI expects node files to be in the main directory, not in a subfolder. 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
56 lines
3.1 KiB
Python
56 lines
3.1 KiB
Python
from typing import Optional, Union
|
|
from pydantic import BaseModel, Field
|
|
from ..utils import BaseRequest, normalization_loras # normalization_loras is not used in this specific class but kept for consistency with the example if other similar classes might need it.
|
|
|
|
|
|
# link api json(step1x-edit.json)
|
|
class Step1xEdit(BaseRequest):
|
|
"""
|
|
Request model for the Step1X Edit API.
|
|
|
|
Step1X-Edit transforms your photos with simple instructions into stunning, professional-quality edits—rivaling top proprietary tools.
|
|
"""
|
|
prompt: str = Field(..., description="The prompt to generate an image from.")
|
|
image: str = Field(..., description="The image URL to generate an image from. Needs to match the dimensions of the mask.")
|
|
negative_prompt: Optional[str] = Field(
|
|
default="",
|
|
description=
|
|
"\n The negative prompt to use. Use it to address details that you don't want\n in the image. This could be colors, objects, scenery and even the small details\n (e.g. moustache, blurry, low resolution).\n "
|
|
)
|
|
seed: Optional[int] = Field(
|
|
default=-1, description="\n The same seed and the same prompt given to the same version of the model\n will output the same image every time.\n "
|
|
) # JSON has no default, using None as it's optional. Example used -1.
|
|
guidance_scale: Optional[float] = Field(
|
|
default=4.0,
|
|
ge=0,
|
|
le=20,
|
|
description="\n The CFG (Classifier Free Guidance) scale is a measure of how close you want\n the model to stick to your prompt when looking for a related image to show you.\n ")
|
|
num_inference_steps: Optional[int] = Field(default=30, ge=1, le=50, description="The number of inference steps to perform.")
|
|
enable_safety_checker: Optional[bool] = Field(
|
|
default=True, description="If set to true, the safety checker will be enabled.") # JSON notes 'disabled: true' for this field, which might mean it's not user-configurable through this schema or always true.
|
|
|
|
def build_payload(self) -> dict:
|
|
"""Builds the request payload dictionary."""
|
|
payload = {
|
|
"prompt": self.prompt,
|
|
"image": self.image,
|
|
"negative_prompt": self.negative_prompt,
|
|
"seed": self.seed,
|
|
"guidance_scale": self.guidance_scale,
|
|
"num_inference_steps": self.num_inference_steps,
|
|
"enable_safety_checker": self.enable_safety_checker,
|
|
}
|
|
return self._remove_empty_fields(payload)
|
|
|
|
def get_api_path(self):
|
|
"""Gets the API path for the request. Corresponds to api_path in the interface configuration json"""
|
|
return "/api/v3/wavespeed-ai/step1x-edit"
|
|
|
|
def field_required(self):
|
|
"""Corresponds to required fields in the interface configuration json"""
|
|
return ["prompt", "image"]
|
|
|
|
def field_order(self):
|
|
"""Corresponds to x-fal-order-properties in the interface configuration json"""
|
|
return ["prompt", "image", "negative_prompt", "seed", "guidance_scale", "num_inference_steps", "enable_safety_checker"]
|