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

“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 = 100
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 = None
114
  encodings = tokenizer(
115
  prompt,
116
  padding="longest",