diff --git a/data.py b/data.py index a3beea24..e77f76f0 100644 --- a/data.py +++ b/data.py @@ -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"])