Save HFToken only if it is present

This commit is contained in:
Nicholas Koh 2023-01-19 14:41:41 +08:00 committed by Kent Keirsey
parent d9fa505412
commit a8bb1a1109

View File

@ -676,7 +676,8 @@ def download_weights(opt:dict) -> Union[str, None]:
return
access_token = authenticate()
HfFolder.save_token(access_token)
if access_token is not None:
HfFolder.save_token(access_token)
print('\n** DOWNLOADING WEIGHTS **')
successfully_downloaded = download_weight_datasets(models, access_token, precision=precision)