22 lines
416 B
Python
22 lines
416 B
Python
import os
|
|
|
|
|
|
def login(token):
|
|
from huggingface_hub import login
|
|
import time
|
|
|
|
while True:
|
|
try:
|
|
login(token)
|
|
print('HF login ok.')
|
|
break
|
|
except Exception as e:
|
|
print(f'HF login failed: {e}. Retrying')
|
|
time.sleep(0.5)
|
|
|
|
|
|
hf_token = os.environ.get('HF_TOKEN', None)
|
|
|
|
if hf_token is not None:
|
|
login(hf_token)
|