Update README.md

README.md

@@ -14,34 +14,35 @@ datasets:
 - eth-dl-rewards/math-problems-for-sft
 ---
 
-# Model Trained Using AutoTrain
-
-This model was trained using AutoTrain. For more information, please visit [AutoTrain](https://hf.co/docs/autotrain).
+# The M is for Math.
 
 # Usage
 
 ```python
 
 from transformers import AutoModelForCausalLM, AutoTokenizer
+import torch
 
-model_path = "PATH_TO_THIS_REPO"
+model_path = "bfuzzy1/acheron-m"
 
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 model = AutoModelForCausalLM.from_pretrained(
     model_path,
     device_map="auto",
-    torch_dtype='auto'
-).eval()
+    torch_dtype='auto',
+    trust_remote_code=True
+)
 
-# Prompt content: "hi"
 messages = [
-    {"role": "user", "content": "hi"}
+    {"role": "user", "content": "What's 2 + 2 -3?"}
 ]
 
 input_ids = tokenizer.apply_chat_template(conversation=messages, tokenize=True, add_generation_prompt=True, return_tensors='pt')
-output_ids = model.generate(input_ids.to('cuda'))
+output_ids = model.generate(
+    input_ids.to('mps' if torch.backends.mps.is_available() else 'cpu'),
+    max_new_tokens=100
+)
 response = tokenizer.decode(output_ids[0][input_ids.shape[1]:], skip_special_tokens=True)
 
-# Model response: "Hello! How can I assist you today?"
 print(response)
 ```