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_image26    27    # switch to "mps" for apple devices28    pipe = DiffusionPipeline.from_pretrained("iitolstykh/VIBE-Image-Edit", dtype=torch.bfloat16, device_map="cuda")29    30    prompt = "Turn this cat into a dog"31    input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png")32    33    image = pipe(image=input_image, prompt=prompt).images[0]34    with open('iitolstykh_VIBE-Image-Edit_0.txt', 'w', encoding='utf-8') as f:35        f.write('Everything was good in iitolstykh_VIBE-Image-Edit_0.txt')36except Exception as e:37    import os38    from slack_sdk import WebClient39    client = WebClient(token=os.environ['SLACK_TOKEN'])40    client.chat_postMessage(41        channel='#hub-model-metadata-snippets-sprint',42        text='Problem in <https://huggingface.co/datasets/model-metadata/code_execution_files/blob/main/iitolstykh_VIBE-Image-Edit_0.txt|iitolstykh_VIBE-Image-Edit_0.txt>',43    )44 45    with open('iitolstykh_VIBE-Image-Edit_0.txt', 'a', encoding='utf-8') as f:46        import traceback47        f.write('''```CODE: 48import torch49from diffusers import DiffusionPipeline50from diffusers.utils import load_image51 52# switch to "mps" for apple devices53pipe = DiffusionPipeline.from_pretrained("iitolstykh/VIBE-Image-Edit", dtype=torch.bfloat16, device_map="cuda")54 55prompt = "Turn this cat into a dog"56input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png")57 58image = pipe(image=input_image, prompt=prompt).images[0]59```60 61ERROR: 62''')63        traceback.print_exc(file=f)64    65finally:66    from huggingface_hub import upload_file67    upload_file(68        path_or_fileobj='iitolstykh_VIBE-Image-Edit_0.txt',69        repo_id='model-metadata/code_execution_files',70        path_in_repo='iitolstykh_VIBE-Image-Edit_0.txt',71        repo_type='dataset',72    )73