MiniMaxAI_MiniMax-M2_0.py

# /// script
# requires-python = ">=3.12"
# dependencies = [
#     "numpy",
#     "einops",
#     "pandas",
#     "matplotlib",
#     "protobuf",
#     "torch",
#     "sentencepiece",
#     "torchvision",
#     "transformers",
#     "timm",
#     "diffusers",
#     "sentence-transformers",
#     "accelerate",
#     "peft",
#     "slack-sdk",
# ]
# ///

try:
    # Use a pipeline as a high-level helper
    from transformers import pipeline
    
    pipe = pipeline("text-generation", model="MiniMaxAI/MiniMax-M2", trust_remote_code=True)
    messages = [
        {"role": "user", "content": "Who are you?"},
    ]
    pipe(messages)
    with open('MiniMaxAI_MiniMax-M2_0.txt', 'w', encoding='utf-8') as f:
        f.write('Everything was good in MiniMaxAI_MiniMax-M2_0.txt')
except Exception as e:
    import os
    from slack_sdk import WebClient
    client = WebClient(token=os.environ['SLACK_TOKEN'])
    client.chat_postMessage(
        channel='#hub-model-metadata-snippets-sprint',
        text='Problem in <https://huggingface.co/datasets/model-metadata/code_execution_files/blob/main/MiniMaxAI_MiniMax-M2_0.txt|MiniMaxAI_MiniMax-M2_0.txt>',
    )

    with open('MiniMaxAI_MiniMax-M2_0.txt', 'a', encoding='utf-8') as f:
        import traceback
        f.write('''```CODE: 
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="MiniMaxAI/MiniMax-M2", trust_remote_code=True)
messages = [
    {"role": "user", "content": "Who are you?"},
]
pipe(messages)
```

ERROR: 
''')
        traceback.print_exc(file=f)
    
finally:
    from huggingface_hub import upload_file
    upload_file(
        path_or_fileobj='MiniMaxAI_MiniMax-M2_0.txt',
        repo_id='model-metadata/code_execution_files',
        path_in_repo='MiniMaxAI_MiniMax-M2_0.txt',
        repo_type='dataset',
    )