Update trl-lora.py
Browse files- trl-lora.py +1 -1
trl-lora.py
CHANGED
|
@@ -64,7 +64,7 @@ trainer = SFTTrainer(
|
|
| 64 |
tokenizer=tokenizer,
|
| 65 |
packing=True,
|
| 66 |
dataset_kwargs={
|
| 67 |
-
"add_special_tokens":
|
| 68 |
"append_concat_token": False, # make sure to not add additional tokens when packing
|
| 69 |
}
|
| 70 |
)
|
|
|
|
| 64 |
tokenizer=tokenizer,
|
| 65 |
packing=True,
|
| 66 |
dataset_kwargs={
|
| 67 |
+
"add_special_tokens": False, # <bos> and <eos> should be part of the dataset.
|
| 68 |
"append_concat_token": False, # make sure to not add additional tokens when packing
|
| 69 |
}
|
| 70 |
)
|