model-metadata/code_python_files

1# /// script2# requires-python = ">=3.12"3# dependencies = [4#     "numpy",5#     "einops",6#     "pandas",7#     "matplotlib",8#     "protobuf",9#     "torch",10#     "sentencepiece",11#     "torchvision",12#     "transformers",13#     "timm",14#     "diffusers",15#     "sentence-transformers",16#     "accelerate",17#     "peft",18#     "slack-sdk",19# ]20# ///21 22try:23    import torch24    from diffusers import DiffusionPipeline25    from diffusers.utils import load_image, export_to_video26    27    # switch to "mps" for apple devices28    pipe = DiffusionPipeline.from_pretrained("Video-Reason/VBVR-Wan2.2", dtype=torch.bfloat16, device_map="cuda")29    pipe.to("cuda")30    31    prompt = "A man with short gray hair plays a red electric guitar."32    image = load_image(33        "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/guitar-man.png"34    )35    36    output = pipe(image=image, prompt=prompt).frames[0]37    export_to_video(output, "output.mp4")38    with open('Video-Reason_VBVR-Wan2.2_0.txt', 'w', encoding='utf-8') as f:39        f.write('Everything was good in Video-Reason_VBVR-Wan2.2_0.txt')40except Exception as e:41    import os42    from slack_sdk import WebClient43    client = WebClient(token=os.environ['SLACK_TOKEN'])44    client.chat_postMessage(45        channel='#hub-model-metadata-snippets-sprint',46        text='Problem in <https://huggingface.co/datasets/model-metadata/code_execution_files/blob/main/Video-Reason_VBVR-Wan2.2_0.txt|Video-Reason_VBVR-Wan2.2_0.txt>',47    )48 49    with open('Video-Reason_VBVR-Wan2.2_0.txt', 'a', encoding='utf-8') as f:50        import traceback51        f.write('''```CODE: 52import torch53from diffusers import DiffusionPipeline54from diffusers.utils import load_image, export_to_video55 56# switch to "mps" for apple devices57pipe = DiffusionPipeline.from_pretrained("Video-Reason/VBVR-Wan2.2", dtype=torch.bfloat16, device_map="cuda")58pipe.to("cuda")59 60prompt = "A man with short gray hair plays a red electric guitar."61image = load_image(62    "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/guitar-man.png"63)64 65output = pipe(image=image, prompt=prompt).frames[0]66export_to_video(output, "output.mp4")67```68 69ERROR: 70''')71        traceback.print_exc(file=f)72    73finally:74    from huggingface_hub import upload_file75    upload_file(76        path_or_fileobj='Video-Reason_VBVR-Wan2.2_0.txt',77        repo_id='model-metadata/code_execution_files',78        path_in_repo='Video-Reason_VBVR-Wan2.2_0.txt',79        repo_type='dataset',80    )81