qnguyen3 commited on
Commit
99e6d63
1 Parent(s): 786b049

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -103,7 +103,14 @@ def bot_streaming(message, history):
103
  streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
104
 
105
  image_tensor = model.process_images([image], model.config).to(dtype=model.dtype)
106
- generation_kwargs = dict(input_ids=input_ids.to('cuda'), images=image_tensor.to('cuda'), streamer=streamer, max_new_tokens=512, stopping_criteria=[stopping_criteria], temperature=0.2)
 
 
 
 
 
 
 
107
  generated_text = ""
108
  thread = Thread(target=model.generate, kwargs=generation_kwargs)
109
  thread.start()
 
103
  streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
104
 
105
  image_tensor = model.process_images([image], model.config).to(dtype=model.dtype)
106
+ generation_kwargs = dict(input_ids=input_ids.to('cuda'),
107
+ images=image_tensor.to('cuda'),
108
+ streamer=streamer, max_new_tokens=512,
109
+ stopping_criteria=[stopping_criteria],
110
+ temperature=0.25,
111
+ repetition_penalty=1.05,
112
+ top_k=40,
113
+ top_p=0.95)
114
  generated_text = ""
115
  thread = Thread(target=model.generate, kwargs=generation_kwargs)
116
  thread.start()