HFswapnil commited on
Commit
d4eb41b
·
verified ·
1 Parent(s): 583e3cc

Update run_model.py

Browse files
Files changed (1) hide show
  1. 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 = mode.generate(**inputs, **generation_args)
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"]