Update app.py
Browse files
app.py
CHANGED
@@ -43,7 +43,7 @@ outputs = vl_gpt.language_model.generate(
|
|
43 |
pad_token_id=tokenizer.eos_token_id,
|
44 |
bos_token_id=tokenizer.bos_token_id,
|
45 |
eos_token_id=tokenizer.eos_token_id,
|
46 |
-
max_new_tokens=
|
47 |
do_sample=False,
|
48 |
use_cache=True
|
49 |
)
|
|
|
43 |
pad_token_id=tokenizer.eos_token_id,
|
44 |
bos_token_id=tokenizer.bos_token_id,
|
45 |
eos_token_id=tokenizer.eos_token_id,
|
46 |
+
max_new_tokens=128,
|
47 |
do_sample=False,
|
48 |
use_cache=True
|
49 |
)
|