chore: Revert back to token

alexandrainst · Jul 31, 2023 · 47c5494 · 47c5494
1 parent a414038
commit 47c5494
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/coral_models/data.py b/src/coral_models/data.py
@@ -29,7 +29,7 @@ def load_data(cfg: DictConfig) -> DatasetDict | IterableDatasetDict:
     dataset = load_dataset(
         path=cfg.dataset.id,
         name=cfg.dataset.subset,
-        use_auth_token=os.getenv("HUGGINGFACE_HUB_TOKEN", True),
+        token=os.getenv("HUGGINGFACE_HUB_TOKEN", True),
         streaming=True,
     )