Skip to content

Commit 3ea242c

Browse files
authored
fix typo in trainer.py (#81)
seq_lenth -> seq_length
1 parent 7bdfa85 commit 3ea242c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

training/trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -178,7 +178,7 @@ def train(
178178
# model is used. The default model uses n_positions. If no config settings can be found just default
179179
# to 1024 as this is probably supported by most models.
180180
conf = model.config
181-
max_length: int = getattr(conf, "n_positions", getattr(conf, "seq_lenth", 1024))
181+
max_length: int = getattr(conf, "n_positions", getattr(conf, "seq_length", 1024))
182182

183183
processed_dataset = preprocess_dataset(tokenizer=tokenizer, max_length=max_length, seed=seed)
184184

0 commit comments

Comments
 (0)