model-metadata/code_python_files

1# /// script2# requires-python = ">=3.12"3# dependencies = [4#     "numpy",5#     "einops",6#     "pandas",7#     "protobuf",8#     "torch",9#     "sentencepiece",10#     "torchvision",11#     "transformers",12#     "timm",13#     "diffusers",14#     "sentence-transformers",15#     "accelerate",16#     "peft",17#     "slack-sdk",18# ]19# ///20 21try:22    # Load model directly23    from transformers import AutoModelForCausalLM24    model = AutoModelForCausalLM.from_pretrained("ByteDance/Ouro-1.4B", trust_remote_code=True, torch_dtype="auto")25    with open('ByteDance_Ouro-1.4B_1.txt', 'w', encoding='utf-8') as f:26        f.write('Everything was good in ByteDance_Ouro-1.4B_1.txt')27except Exception as e:28    import os29    from slack_sdk import WebClient30    client = WebClient(token=os.environ['SLACK_TOKEN'])31    client.chat_postMessage(32        channel='#hub-model-metadata-snippets-sprint',33        text='Problem in <https://huggingface.co/datasets/model-metadata/code_execution_files/blob/main/ByteDance_Ouro-1.4B_1.txt|ByteDance_Ouro-1.4B_1.txt>',34    )35 36    with open('ByteDance_Ouro-1.4B_1.txt', 'a', encoding='utf-8') as f:37        import traceback38        f.write('''```CODE: 39# Load model directly40from transformers import AutoModelForCausalLM41model = AutoModelForCausalLM.from_pretrained("ByteDance/Ouro-1.4B", trust_remote_code=True, torch_dtype="auto")42```43 44ERROR: 45''')46        traceback.print_exc(file=f)47    48finally:49    from huggingface_hub import upload_file50    upload_file(51        path_or_fileobj='ByteDance_Ouro-1.4B_1.txt',52        repo_id='model-metadata/code_execution_files',53        path_in_repo='ByteDance_Ouro-1.4B_1.txt',54        repo_type='dataset',55    )56