model-metadata/code_python_files

1# /// script2# requires-python = ">=3.12"3# dependencies = [4#     "torch",5#     "torchvision",6#     "transformers",7#     "accelerate",8#     "peft",9#     "slack-sdk",10# ]11# ///12 13try:14    # Load model directly15    from transformers import AutoModel16    model = AutoModel.from_pretrained("Open-Bee/Bee-8B-RL", trust_remote_code=True, torch_dtype="auto")17    with open('Open-Bee_Bee-8B-RL_1.txt', 'w', encoding='utf-8') as f:18        f.write('Everything was good in Open-Bee_Bee-8B-RL_1.txt')19except Exception as e:20    import os21    from slack_sdk import WebClient22    client = WebClient(token=os.environ['SLACK_TOKEN'])23    client.chat_postMessage(24        channel='#exp-slack-alerts',25        text='Problem in <https://huggingface.co/datasets/model-metadata/code_execution_files/blob/main/Open-Bee_Bee-8B-RL_1.txt|Open-Bee_Bee-8B-RL_1.txt>',26    )27 28    with open('Open-Bee_Bee-8B-RL_1.txt', 'a', encoding='utf-8') as f:29        import traceback30        f.write('''```CODE: 31# Load model directly32from transformers import AutoModel33model = AutoModel.from_pretrained("Open-Bee/Bee-8B-RL", trust_remote_code=True, torch_dtype="auto")34```35 36ERROR: 37''')38        traceback.print_exc(file=f)39    40finally:41    from huggingface_hub import upload_file42    upload_file(43        path_or_fileobj='Open-Bee_Bee-8B-RL_1.txt',44        repo_id='model-metadata/code_execution_files',45        path_in_repo='Open-Bee_Bee-8B-RL_1.txt',46        repo_type='dataset',47    )48