Update README.md
Browse files
README.md
CHANGED
|
@@ -66,7 +66,8 @@ model = MFuyuForCausalLM.from_pretrained("TIGER-Lab/Mantis-8B-Fuyu", device_map=
|
|
| 66 |
generation_kwargs = {
|
| 67 |
"max_new_tokens": 1024,
|
| 68 |
"num_beams": 1,
|
| 69 |
-
"do_sample": False
|
|
|
|
| 70 |
}
|
| 71 |
|
| 72 |
# chat
|
|
|
|
| 66 |
generation_kwargs = {
|
| 67 |
"max_new_tokens": 1024,
|
| 68 |
"num_beams": 1,
|
| 69 |
+
"do_sample": False,
|
| 70 |
+
"pad_token_id": processor.tokenizer.eos_token_id,
|
| 71 |
}
|
| 72 |
|
| 73 |
# chat
|