Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ def run_inference(message, history):
51
  images = encoded.images
52
  tokens = encoded.tokens
53
 
54
- out_tokens, _ = generate([tokens], model, images=[images], max_tokens=1024, temperature=0.45, eos_id=tokenizer.instruct_tokenizer.tokenizer.eos_id)
55
  result = tokenizer.decode(out_tokens[0])
56
  return result
57
 
 
51
  images = encoded.images
52
  tokens = encoded.tokens
53
 
54
+ out_tokens, _ = generate([tokens], model, images=[images], max_tokens=512, temperature=0.45, eos_id=tokenizer.instruct_tokenizer.tokenizer.eos_id)
55
  result = tokenizer.decode(out_tokens[0])
56
  return result
57