Ravi theja K commited on
Commit
3cb9dc4
·
verified ·
1 Parent(s): 02a638a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ model = AutoModelForCausalLM.from_pretrained(
24
 
25
  content = "5x + 35 = 7x - 60 + 10. Solve for x"
26
  messages = [{"role": "user", "content": content}]
27
- input_ids = Llamatokenizertokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt").to("cuda")
28
 
29
  outputs = model.generate(input_ids=input_ids, max_new_tokens=256)
30
  print(Llamatokenizer.decode(outputs[0]))
 
24
 
25
  content = "5x + 35 = 7x - 60 + 10. Solve for x"
26
  messages = [{"role": "user", "content": content}]
27
+ input_ids = Llamatokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt").to("cuda")
28
 
29
  outputs = model.generate(input_ids=input_ids, max_new_tokens=256)
30
  print(Llamatokenizer.decode(outputs[0]))