model-metadata/code_python_files

1# /// script2# requires-python = ">=3.12"3# dependencies = [4#     "torch",5#     "torchvision",6#     "transformers",7#     "diffusers",8#     "sentence-transformers",9#     "accelerate",10#     "peft",11#     "slack-sdk",12# ]13# ///14 15try:16    import torch17    from diffusers import DiffusionPipeline18    from diffusers.utils import load_image, export_to_video19    20    pipe = DiffusionPipeline.from_pretrained("vita-video-gen/svi-model", torch_dtype=torch.float16)21    pipe.to("cuda")22    23    prompt = "A man with short gray hair plays a red electric guitar."24    image = load_image(25        "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/guitar-man.png"26    )27    28    output = pipe(image=image, prompt=prompt).frames[0]29    export_to_video(output, "output.mp4")30    with open('vita-video-gen_svi-model_0.txt', 'w', encoding='utf-8') as f:31        f.write('Everything was good in vita-video-gen_svi-model_0.txt')32except Exception as e:33    import os34    from slack_sdk import WebClient35    client = WebClient(token=os.environ['SLACK_TOKEN'])36    client.chat_postMessage(37        channel='#hub-model-metadata-snippets-sprint',38        text='Problem in <https://huggingface.co/datasets/model-metadata/code_execution_files/blob/main/vita-video-gen_svi-model_0.txt|vita-video-gen_svi-model_0.txt>',39    )40 41    with open('vita-video-gen_svi-model_0.txt', 'a', encoding='utf-8') as f:42        import traceback43        f.write('''```CODE: 44import torch45from diffusers import DiffusionPipeline46from diffusers.utils import load_image, export_to_video47 48pipe = DiffusionPipeline.from_pretrained("vita-video-gen/svi-model", torch_dtype=torch.float16)49pipe.to("cuda")50 51prompt = "A man with short gray hair plays a red electric guitar."52image = load_image(53    "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/guitar-man.png"54)55 56output = pipe(image=image, prompt=prompt).frames[0]57export_to_video(output, "output.mp4")58```59 60ERROR: 61''')62        traceback.print_exc(file=f)63    64finally:65    from huggingface_hub import upload_file66    upload_file(67        path_or_fileobj='vita-video-gen_svi-model_0.txt',68        repo_id='model-metadata/code_execution_files',69        path_in_repo='vita-video-gen_svi-model_0.txt',70        repo_type='dataset',71    )72