Update README.md

README.md

@@ -42,7 +42,7 @@ In the table, all models are fine-tuned based on the Qwen2.5-Math-7B base model.
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model_name = "Qwen/Qwen2.5-Math-7B-Instruct"
+model_name = "SunnyLin/Qwen2.5-7B-DPO-VP"
 device = "cuda" # the device to load the model onto
 
 model = AutoModelForCausalLM.from_pretrained(
@@ -68,7 +68,7 @@ model_inputs = tokenizer([text], return_tensors="pt").to(device)
 
 generated_ids = model.generate(
     **model_inputs,
-    max_new_tokens=512
+    max_new_tokens=2048
 )
 generated_ids = [
     output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)