INNER CODE UNIT · Python
_hf_login
Future-House/BixBench · generate_zeroshot_evals.py:25
def _hf_login():
HF_TOKEN = os.getenv("HF_TOKEN")
if HF_TOKEN is None:
logger.warning(
"HF_TOKEN environment variable not found. Please set this in your environment "
"or create a .env file with this value to enable Hugging Face model access."
)
else:
try:
login(token=HF_TOKEN)
except Exception:
logger.exception("Failed to login with HF_TOKEN")
def parse_args():
parser = argparse.ArgumentParser(
description="BixBench Zero-ShotEvaluation Tool",
formatter_class=argparse.ArgumentDefaultsHelpFormatter,