Skip to content

Commit ed3dbea

Browse files
committed
SEP is only defined for HFTokenizer
1 parent 90ee34c commit ed3dbea

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_training.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -649,8 +649,8 @@ def test_pretrain_shared_t5_mlm(self):
649649
--checkpoint-activations
650650
--exit-interval {exit_interval}
651651
652-
--merge-file {data_dir}/gpt2-tiny-merges.txt
653-
--vocab-file {data_dir}/gpt2-tiny-vocab.json
652+
--tokenizer-type PretrainedFromHF
653+
--tokenizer-name-or-path gpt2
654654
--log-path {logs_dir}
655655
--save {output_dir}/checkpoints
656656
--load {output_dir}/checkpoints

0 commit comments

Comments
 (0)