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 AutoModelForCausalLM16    model = AutoModelForCausalLM.from_pretrained("inclusionAI/LLaDA2.0-flash-preview", trust_remote_code=True, torch_dtype="auto")17    with open('inclusionAI_LLaDA2.0-flash-preview_2.txt', 'w', encoding='utf-8') as f:18        f.write('Everything was good in inclusionAI_LLaDA2.0-flash-preview_2.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/inclusionAI_LLaDA2.0-flash-preview_2.txt|inclusionAI_LLaDA2.0-flash-preview_2.txt>',26    )27 28    with open('inclusionAI_LLaDA2.0-flash-preview_2.txt', 'a', encoding='utf-8') as f:29        import traceback30        f.write('''```CODE: 31# Load model directly32from transformers import AutoModelForCausalLM33model = AutoModelForCausalLM.from_pretrained("inclusionAI/LLaDA2.0-flash-preview", 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='inclusionAI_LLaDA2.0-flash-preview_2.txt',44        repo_id='model-metadata/code_execution_files',45        path_in_repo='inclusionAI_LLaDA2.0-flash-preview_2.txt',46        repo_type='dataset',47    )48