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",
|