model-metadata/code_python_files

1# /// script2# requires-python = ">=3.12"3# dependencies = [4#     "numpy",5#     "einops",6#     "pandas",7#     "protobuf",8#     "torch",9#     "torchvision",10#     "transformers",11#     "timm",12#     "diffusers",13#     "sentence-transformers",14#     "accelerate",15#     "peft",16#     "slack-sdk",17# ]18# ///19 20try:21    # Load model directly22    from transformers import AutoModelForCausalLM23    model = AutoModelForCausalLM.from_pretrained("BAAI/Emu3.5-Image", dtype="auto")24    with open('BAAI_Emu3.5-Image_0.txt', 'w', encoding='utf-8') as f:25        f.write('Everything was good in BAAI_Emu3.5-Image_0.txt')26except Exception as e:27    import os28    from slack_sdk import WebClient29    client = WebClient(token=os.environ['SLACK_TOKEN'])30    client.chat_postMessage(31        channel='#hub-model-metadata-snippets-sprint',32        text='Problem in <https://huggingface.co/datasets/model-metadata/code_execution_files/blob/main/BAAI_Emu3.5-Image_0.txt|BAAI_Emu3.5-Image_0.txt>',33    )34 35    with open('BAAI_Emu3.5-Image_0.txt', 'a', encoding='utf-8') as f:36        import traceback37        f.write('''```CODE: 38# Load model directly39from transformers import AutoModelForCausalLM40model = AutoModelForCausalLM.from_pretrained("BAAI/Emu3.5-Image", dtype="auto")41```42 43ERROR: 44''')45        traceback.print_exc(file=f)46    47finally:48    from huggingface_hub import upload_file49    upload_file(50        path_or_fileobj='BAAI_Emu3.5-Image_0.txt',51        repo_id='model-metadata/code_execution_files',52        path_in_repo='BAAI_Emu3.5-Image_0.txt',53        repo_type='dataset',54    )55