fix: Default token to True in train_ngram_model

alexandrainst · Jul 31, 2023 · a414038 · a414038
1 parent f611731
commit a414038
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/scripts/train_ngram_decoder.py b/src/scripts/train_ngram_decoder.py
@@ -33,7 +33,7 @@ def train_ngram_model(cfg: DictConfig) -> None:
         path=cfg.model.decoder.dataset_id,
         name=cfg.model.decoder.dataset_subset,
         split=cfg.model.decoder.dataset_split,
-        use_auth_token=os.getenv("HUGGINGFACE_HUB_TOKEN"),
+        token=os.getenv("HUGGINGFACE_HUB_TOKEN", True),
     )
     assert isinstance(dataset, Dataset)