Spaces:
Runtime error
Runtime error
Li
commited on
Commit
•
5545c07
1
Parent(s):
687c6ab
“update”
Browse files
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 =
|
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",
|