Arturo Jiménez de los Galanes Reguillos commited on
Commit
e9017bf
·
1 Parent(s): 47728bf

Add input_ids key to generation_kwargs

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ def translate(python):
41
  input_ids = inputs.input_ids
42
 
43
  generation_kwargs = dict(
44
- input_ids,
45
  attention_mask=attention_mask,
46
  max_new_tokens=1024,
47
  do_sample=False,
 
41
  input_ids = inputs.input_ids
42
 
43
  generation_kwargs = dict(
44
+ input_ids=input_ids,
45
  attention_mask=attention_mask,
46
  max_new_tokens=1024,
47
  do_sample=False,