chris-propeller commited on
Commit
bd998ab
·
verified ·
1 Parent(s): 1c15af1

Allow more tokens

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ def describe_image(image, question):
34
  # Generate output using the model on GPU
35
  output = model.generate_from_batch(
36
  inputs,
37
- GenerationConfig(max_new_tokens=1024, stop_strings="<|endoftext|>"),
38
  tokenizer=processor.tokenizer,
39
  )
40
 
 
34
  # Generate output using the model on GPU
35
  output = model.generate_from_batch(
36
  inputs,
37
+ GenerationConfig(max_new_tokens=2048, stop_strings="<|endoftext|>"),
38
  tokenizer=processor.tokenizer,
39
  )
40