Spaces:
Running
on
L40S
Running
on
L40S
Update app.py
Browse files
app.py
CHANGED
@@ -131,9 +131,9 @@ def vision_language_understanding(image, text):
|
|
131 |
|
132 |
# Prepare hyperparameters
|
133 |
GENERATION_CONFIG = GenerationConfig(
|
134 |
-
pad_token_id=
|
135 |
-
bos_token_id=
|
136 |
-
eos_token_id=
|
137 |
max_new_tokens=320,
|
138 |
)
|
139 |
|
|
|
131 |
|
132 |
# Prepare hyperparameters
|
133 |
GENERATION_CONFIG = GenerationConfig(
|
134 |
+
pad_token_id=tokenizer.pad_token_id,
|
135 |
+
bos_token_id=tokenizer.bos_token_id,
|
136 |
+
eos_token_id=tokenizer.eos_token_id,
|
137 |
max_new_tokens=320,
|
138 |
)
|
139 |
|