chore: Try using use_auth_token instead of token

alexandrainst · Jul 31, 2023 · f611731 · f611731
1 parent d1e17dd
commit f611731
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,
-        token=os.getenv("HUGGINGFACE_HUB_TOKEN", True),
+        use_auth_token=os.getenv("HUGGINGFACE_HUB_TOKEN", True),
         streaming=True,
     )