-
Notifications
You must be signed in to change notification settings - Fork 34
Description
when I use "Change specific numbers:" to augment problem using internlm2-math-20b to, the model answering the question instead of augment it~
import torch
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("internLLMATH", trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained("internLLMATH", trust_remote_code=True, torch_dtype=torch.float16).cuda()
model = model.eval()
response, history = model.chat(tokenizer, "Change specific numbers: Find the center of the circle with equation
print(response)
We complete the square for both
So, the center of the circle is
The answer is: (3,-1).