ByteDance_Video-As-Prompt-Wan2.1-14B_2.py

# /// script
# requires-python = ">=3.12"
# dependencies = [
#     "torch",
#     "torchvision",
#     "transformers",
#     "accelerate",
#     "peft",
#     "slack-sdk",
# ]
# ///

try:
    import torch
    from diffusers import DiffusionPipeline
    from diffusers.utils import load_image, export_to_video
    
    pipe = DiffusionPipeline.from_pretrained("ByteDance/Video-As-Prompt-Wan2.1-14B", torch_dtype=torch.float16)
    pipe.to("cuda")
    
    prompt = "A man with short gray hair plays a red electric guitar."
    image = load_image(
        "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/guitar-man.png"
    )
    
    output = pipe(image=image, prompt=prompt).frames[0]
    export_to_video(output, "output.mp4")
    with open('ByteDance_Video-As-Prompt-Wan2.1-14B_2.txt', 'w', encoding='utf-8') as f:
        f.write('Everything was good in ByteDance_Video-As-Prompt-Wan2.1-14B_2.txt')
except Exception as e:
    import os
    from slack_sdk import WebClient
    client = WebClient(token=os.environ['SLACK_TOKEN'])
    client.chat_postMessage(
        channel='#exp-slack-alerts',
        text='Problem in <https://huggingface.co/datasets/model-metadata/code_execution_files/blob/main/ByteDance_Video-As-Prompt-Wan2.1-14B_2.txt|ByteDance_Video-As-Prompt-Wan2.1-14B_2.txt>',
    )

    with open('ByteDance_Video-As-Prompt-Wan2.1-14B_2.txt', 'a', encoding='utf-8') as f:
        import traceback
        f.write('''```CODE: 
import torch
from diffusers import DiffusionPipeline
from diffusers.utils import load_image, export_to_video

pipe = DiffusionPipeline.from_pretrained("ByteDance/Video-As-Prompt-Wan2.1-14B", torch_dtype=torch.float16)
pipe.to("cuda")

prompt = "A man with short gray hair plays a red electric guitar."
image = load_image(
    "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/guitar-man.png"
)

output = pipe(image=image, prompt=prompt).frames[0]
export_to_video(output, "output.mp4")
```

ERROR: 
''')
        traceback.print_exc(file=f)
    
finally:
    from huggingface_hub import upload_file
    upload_file(
        path_or_fileobj='ByteDance_Video-As-Prompt-Wan2.1-14B_2.txt',
        repo_id='model-metadata/code_execution_files',
        path_in_repo='ByteDance_Video-As-Prompt-Wan2.1-14B_2.txt',
        repo_type='dataset',
    )