init

README.md

@@ -25,7 +25,7 @@ Tabel 2: Model performance on math reasoning tasks. For AIME and AMC, the result
 from transformers import pipeline
 
 question = "math"
-generator = pipeline("text-generation", model="GradientNetwork/Qwen2.5-7B-ECHO-MATH-GRPO", device="cuda")
+generator = pipeline("text-generation", model="GradientResearch/Qwen2.5-7B-ECHO-MATH-GRPO", device="cuda")
 output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
 print(output["generated_text"])
 ```