i2v change from image url to base64 coding

This commit is contained in:
Tuo He
2024-12-05 20:42:55 +00:00
parent 9507ac4932
commit 011dfdf9f2
+3 -1
View File
@@ -45,7 +45,7 @@ class I2VPipelineNode:
return {
"required": {
"prompt": ("STRING", {"default": "add prompt here", "display": "text"}),
"source_image": ("STRING", {"default": "add source image url here", "display": "text"}),
"source_image": ("IMAGE",),
},
"optional": {
"duration": ("INT", {"default": 6, "min": 1, "max": 6, "step": 1, "display": "number"}),
@@ -70,6 +70,8 @@ class I2VPipelineNode:
os.makedirs(directory, exist_ok=True)
output_path = os.path.join(directory, "output-i2v.mp4")
source_image = image2base64(source_image[0])
# Use the user-defined prompt instead of a hardcoded prompt
run_status = get_video_by_i2v_pipeline(prompt, source_image, duration, seed, resolution, is_public, is_enable_prompt_enhancer, guidance_scale, output_path)