Li commited on
Commit
5545c07
1 Parent(s): 687c6ab

“update”

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -110,7 +110,7 @@ def generate(
110
  else:
111
  prompt = [f"<|#image#|>{tokenizer.pad_token * vis_embed_size}<|#endofimage#|>{text.rstrip('.')}"]
112
  bad_words_ids = loc_word_ids
113
- max_generation_length = None
114
  encodings = tokenizer(
115
  prompt,
116
  padding="longest",
 
110
  else:
111
  prompt = [f"<|#image#|>{tokenizer.pad_token * vis_embed_size}<|#endofimage#|>{text.rstrip('.')}"]
112
  bad_words_ids = loc_word_ids
113
+ max_generation_length = 30
114
  encodings = tokenizer(
115
  prompt,
116
  padding="longest",