ucsahin commited on
Commit
20ad0b8
•
1 Parent(s): 24f1fbc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ def bot_streaming(message, history, max_tokens, temperature, top_p, top_k, repet
33
  inputs = processor(text=prompt, images=image, return_tensors="pt").to("cuda:0")
34
 
35
  generation_kwargs = dict(
36
- inputs, streamer=streamer, max_new_tokens=max_tokens,
37
  do_sample=True, temperature=temperature, top_p=top_p,
38
  top_k=top_k, repetition_penalty=repetition_penalty
39
  )
 
33
  inputs = processor(text=prompt, images=image, return_tensors="pt").to("cuda:0")
34
 
35
  generation_kwargs = dict(
36
+ inputs, max_new_tokens=max_tokens,
37
  do_sample=True, temperature=temperature, top_p=top_p,
38
  top_k=top_k, repetition_penalty=repetition_penalty
39
  )