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

Add streamer to generation_kwargs

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