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    # Load model directly17    from transformers import AutoModelForCausalLM18    model = AutoModelForCausalLM.from_pretrained("inclusionAI/Ring-1T", trust_remote_code=True, torch_dtype="auto")19    with open('inclusionAI_Ring-1T_1.txt', 'w', encoding='utf-8') as f:20        f.write('Everything was good in inclusionAI_Ring-1T_1.txt')21except Exception as e:22    import os23    from slack_sdk import WebClient24    client = WebClient(token=os.environ['SLACK_TOKEN'])25    client.chat_postMessage(26        channel='#exp-slack-alerts',27        text='Problem in <https://huggingface.co/datasets/model-metadata/code_execution_files/blob/main/inclusionAI_Ring-1T_1.txt|inclusionAI_Ring-1T_1.txt>',28    )29 30    with open('inclusionAI_Ring-1T_1.txt', 'a', encoding='utf-8') as f:31        import traceback32        f.write('''```CODE: 33# Load model directly34from transformers import AutoModelForCausalLM35model = AutoModelForCausalLM.from_pretrained("inclusionAI/Ring-1T", trust_remote_code=True, torch_dtype="auto")36```37 38ERROR: 39''')40        traceback.print_exc(file=f)41    42finally:43    from huggingface_hub import upload_file44    upload_file(45        path_or_fileobj='inclusionAI_Ring-1T_1.txt',46        repo_id='model-metadata/code_execution_files',47        path_in_repo='inclusionAI_Ring-1T_1.txt',48        repo_type='dataset',49    )50