nodes/tcarwash-ComfyUI_RunpodNodes
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
Files
main
tcarwash-ComfyUI_RunpodNodes/api_examples.py
T

415 lines
17 KiB
Python
Raw Permalink Blame History

# api_examples.py
# RunPod Public Endpoint parameter definitions
# These are hardcoded endpoint IDs and their default parameter examples
# =============================================================================
# IMAGE GENERATION ENDPOINTS
# =============================================================================
FLUX_DEV = {
"endpoint": "black-forest-labs-flux-1-dev",
"display_name": "Flux Dev",
"category": "Image",
"params": {
"prompt": ("STRING", {"default": "A serene mountain landscape at sunset", "multiline": True}),
"negative_prompt": ("STRING", {"default": "", "multiline": True}),
"width": ("INT", {"default": 1024, "min": 256, "max": 1536, "step": 64}),
"height": ("INT", {"default": 1024, "min": 256, "max": 1536, "step": 64}),
"num_inference_steps": ("INT", {"default": 28, "min": 1, "max": 50}),
"guidance": ("FLOAT", {"default": 7.5, "min": 0.0, "max": 10.0, "step": 0.1}),
"seed": ("INT", {"default": -1, "min": -1, "max": 2147483647}),
"image_format": (["png", "jpeg"], {"default": "png"}),
}
}
FLUX_SCHNELL = {
"endpoint": "black-forest-labs-flux-1-schnell",
"display_name": "Flux Schnell",
"category": "Image",
"params": {
"prompt": ("STRING", {"default": "A quick sketch of a mountain", "multiline": True}),
"negative_prompt": ("STRING", {"default": "", "multiline": True}),
"width": ("INT", {"default": 1024, "min": 256, "max": 1536, "step": 64}),
"height": ("INT", {"default": 1024, "min": 256, "max": 1536, "step": 64}),
"num_inference_steps": ("INT", {"default": 4, "min": 1, "max": 8}),
"guidance": ("FLOAT", {"default": 7.5, "min": 0.0, "max": 10.0, "step": 0.1}),
"seed": ("INT", {"default": -1, "min": -1, "max": 2147483647}),
"image_format": (["png", "jpeg"], {"default": "jpeg"}),
}
}
FLUX_KONTEXT_DEV = {
"endpoint": "black-forest-labs-flux-1-kontext-dev",
"display_name": "Flux Kontext Dev (Image Edit)",
"category": "Image",
"params": {
"prompt": ("STRING", {"default": "Edit the image to change the background", "multiline": True}),
"negative_prompt": ("STRING", {"default": "", "multiline": True}),
"image": ("STRING", {"default": "", "multiline": False}), # URL input
"size": ("STRING", {"default": "1024*1024", "multiline": False}),
"num_inference_steps": ("INT", {"default": 28, "min": 1, "max": 50}),
"guidance": ("FLOAT", {"default": 2.0, "min": 0.0, "max": 10.0, "step": 0.1}),
"seed": ("INT", {"default": -1, "min": -1, "max": 2147483647}),
"output_format": (["png", "jpeg"], {"default": "png"}),
"enable_safety_checker": ("BOOLEAN", {"default": True}),
},
"accepts_image": True,
}
QWEN_IMAGE = {
"endpoint": "qwen-image-t2i",
"display_name": "Qwen Image",
"category": "Image",
"params": {
"prompt": ("STRING", {"default": "A fashion-forward woman sitting at cobblestone street in Paris", "multiline": True}),
"negative_prompt": ("STRING", {"default": "", "multiline": True}),
"size": ("STRING", {"default": "1024*1024", "multiline": False}),
"seed": ("INT", {"default": -1, "min": -1, "max": 2147483647}),
"enable_safety_checker": ("BOOLEAN", {"default": True}),
}
}
QWEN_IMAGE_LORA = {
"endpoint": "qwen-image-t2i-lora",
"display_name": "Qwen Image LoRA",
"category": "Image",
"params": {
"prompt": ("STRING", {"default": "Real life Anime in a cozy kitchen", "multiline": True}),
"loras": ("STRING", {"default": "[]", "multiline": True}), # JSON array
"size": ("STRING", {"default": "1024*1024", "multiline": False}),
"seed": ("INT", {"default": -1, "min": -1, "max": 2147483647}),
"enable_safety_checker": ("BOOLEAN", {"default": True}),
}
}
QWEN_IMAGE_EDIT = {
"endpoint": "qwen-image-edit",
"display_name": "Qwen Image Edit",
"category": "Image",
"params": {
"prompt": ("STRING", {"default": "change the color to light grey", "multiline": True}),
"negative_prompt": ("STRING", {"default": "", "multiline": True}),
"image": ("STRING", {"default": "", "multiline": False}),
"seed": ("INT", {"default": -1, "min": -1, "max": 2147483647}),
"output_format": (["png", "jpeg"], {"default": "png"}),
"enable_safety_checker": ("BOOLEAN", {"default": True}),
},
"accepts_image": True,
}
SEEDREAM_3 = {
"endpoint": "seedream-3-0-t2i",
"display_name": "Seedream 3.0",
"category": "Image",
"params": {
"prompt": ("STRING", {"default": "Hyper-realistic photograph of a deep-sea diver", "multiline": True}),
"seed": ("INT", {"default": -1, "min": -1, "max": 2147483647}),
"guidance": ("FLOAT", {"default": 2.0, "min": 0.0, "max": 10.0, "step": 0.1}),
"size": ("STRING", {"default": "1024x1024", "multiline": False}),
}
}
SEEDREAM_4_T2I = {
"endpoint": "seedream-v4-t2i",
"display_name": "Seedream 4.0 T2I",
"category": "Image",
"params": {
"prompt": ("STRING", {"default": "American retro 1950s illustration style", "multiline": True}),
"negative_prompt": ("STRING", {"default": "", "multiline": True}),
"size": ("STRING", {"default": "1024*1024", "multiline": False}),
"seed": ("INT", {"default": -1, "min": -1, "max": 2147483647}),
"enable_safety_checker": ("BOOLEAN", {"default": True}),
}
}
SEEDREAM_4_EDIT = {
"endpoint": "seedream-v4-edit",
"display_name": "Seedream 4.0 Edit",
"category": "Image",
"params": {
"prompt": ("STRING", {"default": "Dress the model in the clothes and hat", "multiline": True}),
"images": ("STRING", {"default": "[]", "multiline": True}), # JSON array of URLs
"size": ("STRING", {"default": "1024*1024", "multiline": False}),
"enable_safety_checker": ("BOOLEAN", {"default": True}),
},
"accepts_image": True,
"num_image_inputs": 2,
}
NANO_BANANA_EDIT = {
"endpoint": "nano-banana-edit",
"display_name": "Nano Banana Edit (Image Combination)",
"category": "Image",
"params": {
"prompt": ("STRING", {"default": "Combine these images into a single realistic scene", "multiline": True}),
"images": ("STRING", {"default": "[]", "multiline": True}), # JSON array of URLs
"enable_safety_checker": ("BOOLEAN", {"default": True}),
},
"accepts_image": True,
"num_image_inputs": 2,
}
# =============================================================================
# VIDEO GENERATION ENDPOINTS
# =============================================================================
INFINITETALK = {
"endpoint": "infinitetalk",
"display_name": "InfiniteTalk",
"category": "Video",
"params": {
"prompt": ("STRING", {"default": "a cartoon computer talking", "multiline": True}),
"image": ("STRING", {"default": "", "multiline": False}),
"audio": ("STRING", {"default": "", "multiline": False}),
"size": (["480p", "720p"], {"default": "480p"}),
"enable_safety_checker": ("BOOLEAN", {"default": True}),
},
"accepts_image": True,
"output_type": "video",
}
KLING_V21_I2V_PRO = {
"endpoint": "kling-v2-1-i2v-pro",
"display_name": "Kling v2.1 I2V Pro",
"category": "Video",
"params": {
"prompt": ("STRING", {"default": "A majestic dragon breathing fire over an ancient castle", "multiline": True}),
"image": ("STRING", {"default": "", "multiline": False}),
"negative_prompt": ("STRING", {"default": "", "multiline": True}),
"guidance_scale": ("FLOAT", {"default": 0.5, "min": 0.0, "max": 1.0, "step": 0.1}),
"duration": ("INT", {"default": 5, "min": 1, "max": 10}),
"enable_safety_checker": ("BOOLEAN", {"default": True}),
},
"accepts_image": True,
"output_type": "video",
}
SEEDANCE_1_PRO = {
"endpoint": "seedance-1-0-pro",
"display_name": "Seedance 1.0 Pro",
"category": "Video",
"params": {
"prompt": ("STRING", {"default": "A pristine Porsche 911 photographed in golden hour lighting", "multiline": True}),
"image": ("STRING", {"default": "", "multiline": False}),
"duration": ("INT", {"default": 5, "min": 1, "max": 10}),
"fps": ("INT", {"default": 24, "min": 12, "max": 60}),
"size": ("STRING", {"default": "1920x1080", "multiline": False}),
},
"accepts_image": True,
"output_type": "video",
}
SORA_2_I2V = {
"endpoint": "sora-2-i2v",
"display_name": "SORA 2 I2V",
"category": "Video",
"params": {
"prompt": ("STRING", {"default": "Action: The character moves slowly forward. Ambient Sound: Distant echoes.", "multiline": True}),
"image": ("STRING", {"default": "", "multiline": False}),
"duration": (["4", "8", "12"], {"default": "4"}),
},
"accepts_image": True,
"output_type": "video",
}
SORA_2_PRO_I2V = {
"endpoint": "sora-2-pro-i2v",
"display_name": "SORA 2 Pro I2V",
"category": "Video",
"params": {
"prompt": ("STRING", {"default": "Action: She opened her hands. Ambient Sound: Soft crackling.", "multiline": True}),
"image": ("STRING", {"default": "", "multiline": False}),
"size": ("STRING", {"default": "720p", "multiline": False}),
"duration": (["4", "8", "12"], {"default": "4"}),
},
"accepts_image": True,
"output_type": "video",
}
WAN_25 = {
"endpoint": "wan-2-5",
"display_name": "WAN 2.5 I2V",
"category": "Video",
"params": {
"prompt": ("STRING", {"default": "A stand-up comedian delivering a dad joke", "multiline": True}),
"image": ("STRING", {"default": "", "multiline": False}),
"negative_prompt": ("STRING", {"default": "", "multiline": True}),
"size": ("STRING", {"default": "1280*720", "multiline": False}),
"duration": ("INT", {"default": 5, "min": 1, "max": 10}),
"seed": ("INT", {"default": -1, "min": -1, "max": 2147483647}),
"enable_prompt_expansion": ("BOOLEAN", {"default": False}),
"enable_safety_checker": ("BOOLEAN", {"default": True}),
},
"accepts_image": True,
"output_type": "video",
}
WAN_22_I2V_720_LORA = {
"endpoint": "wan-2-2-t2v-720-lora",
"display_name": "WAN 2.2 I2V 720p LoRA",
"category": "Video",
"params": {
"prompt": ("STRING", {"default": "orbit 180 around an astronaut on the moon", "multiline": True}),
"image": ("STRING", {"default": "", "multiline": False}),
"high_noise_loras": ("STRING", {"default": "[]", "multiline": True}),
"low_noise_loras": ("STRING", {"default": "[]", "multiline": True}),
"duration": ("INT", {"default": 5, "min": 1, "max": 10}),
"seed": ("INT", {"default": -1, "min": -1, "max": 2147483647}),
"enable_safety_checker": ("BOOLEAN", {"default": True}),
},
"accepts_image": True,
"output_type": "video",
}
WAN_22_I2V_720 = {
"endpoint": "wan-2-2-i2v-720",
"display_name": "WAN 2.2 I2V 720p",
"category": "Video",
"params": {
"prompt": ("STRING", {"default": "cinematic shot: slow-tracking camera glides parallel to a giant white origami boat", "multiline": True}),
"image": ("STRING", {"default": "", "multiline": False}),
"negative_prompt": ("STRING", {"default": "", "multiline": True}),
"size": ("STRING", {"default": "1280*720", "multiline": False}),
"num_inference_steps": ("INT", {"default": 30, "min": 1, "max": 50}),
"guidance": ("FLOAT", {"default": 5.0, "min": 0.0, "max": 10.0, "step": 0.1}),
"duration": ("INT", {"default": 5, "min": 1, "max": 10}),
"flow_shift": ("INT", {"default": 5, "min": 1, "max": 10}),
"seed": ("INT", {"default": -1, "min": -1, "max": 2147483647}),
"enable_prompt_optimization": ("BOOLEAN", {"default": False}),
"enable_safety_checker": ("BOOLEAN", {"default": True}),
},
"accepts_image": True,
"output_type": "video",
}
WAN_22_T2V_720 = {
"endpoint": "wan-2-2-t2v-720",
"display_name": "WAN 2.2 T2V 720p",
"category": "Video",
"params": {
"prompt": ("STRING", {"default": "A serene morning in an ancient forest, golden sunlight filtering through tall pine trees", "multiline": True}),
"negative_prompt": ("STRING", {"default": "", "multiline": True}),
"size": ("STRING", {"default": "1280*720", "multiline": False}),
"num_inference_steps": ("INT", {"default": 30, "min": 1, "max": 50}),
"guidance": ("FLOAT", {"default": 5.0, "min": 0.0, "max": 10.0, "step": 0.1}),
"duration": ("INT", {"default": 5, "min": 1, "max": 10}),
"flow_shift": ("INT", {"default": 5, "min": 1, "max": 10}),
"seed": ("INT", {"default": -1, "min": -1, "max": 2147483647}),
"enable_prompt_optimization": ("BOOLEAN", {"default": False}),
"enable_safety_checker": ("BOOLEAN", {"default": True}),
},
"output_type": "video",
}
WAN_21_I2V_720 = {
"endpoint": "wan-2-1-i2v-720",
"display_name": "WAN 2.1 I2V 720p",
"category": "Video",
"params": {
"prompt": ("STRING", {"default": "The family just took a selfie. They lean in together, smiling and relaxed.", "multiline": True}),
"image": ("STRING", {"default": "", "multiline": False}),
"negative_prompt": ("STRING", {"default": "", "multiline": True}),
"size": ("STRING", {"default": "1280*720", "multiline": False}),
"num_inference_steps": ("INT", {"default": 30, "min": 1, "max": 50}),
"guidance": ("FLOAT", {"default": 5.0, "min": 0.0, "max": 10.0, "step": 0.1}),
"duration": ("INT", {"default": 5, "min": 1, "max": 10}),
"flow_shift": ("INT", {"default": 5, "min": 1, "max": 10}),
"seed": ("INT", {"default": -1, "min": -1, "max": 2147483647}),
"enable_prompt_optimization": ("BOOLEAN", {"default": False}),
"enable_safety_checker": ("BOOLEAN", {"default": True}),
},
"accepts_image": True,
"output_type": "video",
}
WAN_21_T2V_720 = {
"endpoint": "wan-2-1-t2v-720",
"display_name": "WAN 2.1 T2V 720p",
"category": "Video",
"params": {
"prompt": ("STRING", {"default": "Steady rain falls on a bustling Tokyo street at night, neon signs casting vibrant pink and blue light", "multiline": True}),
"negative_prompt": ("STRING", {"default": "", "multiline": True}),
"size": ("STRING", {"default": "1280*720", "multiline": False}),
"num_inference_steps": ("INT", {"default": 30, "min": 1, "max": 50}),
"guidance": ("FLOAT", {"default": 5.0, "min": 0.0, "max": 10.0, "step": 0.1}),
"duration": ("INT", {"default": 5, "min": 1, "max": 10}),
"flow_shift": ("INT", {"default": 5, "min": 1, "max": 10}),
"seed": ("INT", {"default": -1, "min": -1, "max": 2147483647}),
"enable_prompt_optimization": ("BOOLEAN", {"default": False}),
"enable_safety_checker": ("BOOLEAN", {"default": True}),
},
"output_type": "video",
}
# =============================================================================
# AUDIO ENDPOINTS
# =============================================================================
WHISPER_V3_LARGE = {
"endpoint": "whisper-v3-large",
"display_name": "Whisper V3 Large (Speech-to-Text)",
"category": "Audio",
"params": {
"prompt": ("STRING", {"default": "", "multiline": True}),
"audio": ("STRING", {"default": "", "multiline": False}),
},
"output_type": "text",
}
MINIMAX_SPEECH_02_HD = {
"endpoint": "minimax-speech-02-hd",
"display_name": "Minimax Speech 02 HD (Text-to-Speech)",
"category": "Audio",
"params": {
"prompt": ("STRING", {"default": "Welcome to our advanced text-to-speech system", "multiline": True}),
"voice_id": ("STRING", {"default": "Wise_Woman", "multiline": False}),
"speed": ("FLOAT", {"default": 1.0, "min": 0.5, "max": 2.0, "step": 0.1}),
"volume": ("FLOAT", {"default": 1.0, "min": 0.0, "max": 2.0, "step": 0.1}),
"pitch": ("FLOAT", {"default": 0.0, "min": -1.0, "max": 1.0, "step": 0.1}),
"emotion": (["neutral", "happy", "sad", "angry", "fearful", "disgusted", "surprised"], {"default": "neutral"}),
"english_normalization": ("BOOLEAN", {"default": False}),
},
"output_type": "audio",
}
# =============================================================================
# ALL ENDPOINTS REGISTRY
# =============================================================================
ALL_IMAGE_ENDPOINTS = [
FLUX_DEV,
FLUX_SCHNELL,
FLUX_KONTEXT_DEV,
QWEN_IMAGE,
QWEN_IMAGE_LORA,
QWEN_IMAGE_EDIT,
SEEDREAM_3,
SEEDREAM_4_T2I,
SEEDREAM_4_EDIT,
NANO_BANANA_EDIT,
]
ALL_VIDEO_ENDPOINTS = [
INFINITETALK,
KLING_V21_I2V_PRO,
SEEDANCE_1_PRO,
SORA_2_I2V,
SORA_2_PRO_I2V,
WAN_25,
WAN_22_I2V_720_LORA,
WAN_22_I2V_720,
WAN_22_T2V_720,
WAN_21_I2V_720,
WAN_21_T2V_720,
]
ALL_AUDIO_ENDPOINTS = [
WHISPER_V3_LARGE,
MINIMAX_SPEECH_02_HD,
]
ALL_ENDPOINTS = ALL_IMAGE_ENDPOINTS + ALL_VIDEO_ENDPOINTS + ALL_AUDIO_ENDPOINTS
Reference in New Issue View Git Blame Copy Permalink