theostos commited on
Commit
34883a3
·
1 Parent(s): 5e2217f

Fix gen_kwargs, by expanding inputs

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -79,7 +79,7 @@ def generate(term, deps, temperature, top_p, max_new_tokens):
79
  inputs = tokenizer(prompt_text, return_tensors="pt").to(device)
80
  streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
81
  gen_kwargs = dict(
82
- inputs=inputs,
83
  max_new_tokens=int(max_new_tokens),
84
  temperature=float(temperature),
85
  top_p=float(top_p),
 
79
  inputs = tokenizer(prompt_text, return_tensors="pt").to(device)
80
  streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
81
  gen_kwargs = dict(
82
+ **inputs,
83
  max_new_tokens=int(max_new_tokens),
84
  temperature=float(temperature),
85
  top_p=float(top_p),