alan-turing-institute / reginald

Reginald repository for REG Hack Week 23
3 stars 0 forks source link

Add loading of Huggingface token #172

Open rchan26 opened 7 months ago

rchan26 commented 7 months ago

When using some models like Llama2 or Gemma, you need to request access first on Huggingface and then pass in an token argument or sign in via the huggingfce-cli.

For us, it's probably easiest to just require the user to set a HUGGINGFACE_TOKEN environment variable and then pass it to where it's needed (e.g. when we load the Llama2 tokenizer with transformers.AutoTokenizer).