model-metadata/custom_code_py_files

1# /// script2# requires-python = ">=3.12"3# dependencies = [4#     "torch",5#     "torchvision",6#     "transformers",7#     "accelerate",8#     "peft",9# ]10# ///11 12try:13    # Load model directly14    from transformers import AutoTokenizer, AutoModelForCausalLM15    16    tokenizer = AutoTokenizer.from_pretrained("deepseek-ai/DeepSeek-R1", trust_remote_code=True)17    model = AutoModelForCausalLM.from_pretrained("deepseek-ai/DeepSeek-R1", trust_remote_code=True)18    messages = [19        {"role": "user", "content": "Who are you?"},20    ]21    inputs = tokenizer.apply_chat_template(22    	messages,23    	add_generation_prompt=True,24    	tokenize=True,25    	return_dict=True,26    	return_tensors="pt",27    ).to(model.device)28    29    outputs = model.generate(**inputs, max_new_tokens=40)30    print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:]))31    with open('deepseek-ai_DeepSeek-R1_1.txt', 'w') as f:32        f.write('Everything was good in deepseek-ai_DeepSeek-R1_1.txt')33except Exception as e:34    with open('deepseek-ai_DeepSeek-R1_1.txt', 'w') as f:35        import traceback36        traceback.print_exc(file=f)37finally:38    from huggingface_hub import upload_file39    upload_file(40        path_or_fileobj='deepseek-ai_DeepSeek-R1_1.txt',41        repo_id='model-metadata/custom_code_execution_files',42        path_in_repo='deepseek-ai_DeepSeek-R1_1.txt',43        repo_type='dataset',44    )