fix: typo

This commit is contained in:
Zach Nussbaum 2023-04-19 19:36:45 +00:00
parent 6518fa1461
commit 405d8c1bbc

View File

@ -90,7 +90,7 @@ def load_data(config, tokenizer):
# read from huggingface
else:
revison = config["revision"]
revision = config["revision"]
dataset = load_dataset(dataset_path, split="train", revision=revision)
dataset = dataset.train_test_split(test_size=.05, seed=config["seed"])