Spaces:
Sleeping
Sleeping
Update run_model.py
Browse files- run_model.py +1 -1
run_model.py
CHANGED
|
@@ -49,7 +49,7 @@ async def generate_response(prompt: str, context: str = "", history: list = [])
|
|
| 49 |
with torch.no_grad():
|
| 50 |
prompt_text = "\n".join([f"{m['role']}: {m['content']}" for m in message])
|
| 51 |
inputs = tokenizer(prompt_text, return_tensors="pt").to(model.device)
|
| 52 |
-
outputs =
|
| 53 |
outputs = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
| 54 |
|
| 55 |
# return outputs[0]["generated_text"]
|
|
|
|
| 49 |
with torch.no_grad():
|
| 50 |
prompt_text = "\n".join([f"{m['role']}: {m['content']}" for m in message])
|
| 51 |
inputs = tokenizer(prompt_text, return_tensors="pt").to(model.device)
|
| 52 |
+
outputs = model.generate(**inputs, **generation_args)
|
| 53 |
outputs = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
| 54 |
|
| 55 |
# return outputs[0]["generated_text"]
|