Spaces:
Build error
Build error
Ravi theja K
commited on
Update app.py
Browse files
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 =
|
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]))
|