diff --git a/dgmr/hub.py b/dgmr/hub.py index 568883d..821fbf2 100644 --- a/dgmr/hub.py +++ b/dgmr/hub.py @@ -148,7 +148,7 @@ def _from_pretrained( force_download=force_download, proxies=proxies, resume_download=resume_download, - use_auth_token=use_auth_token, + token=use_auth_token, local_files_only=local_files_only, ) model = cls(**model_kwargs["config"])