dpaul93 commited on
Commit
562ef9f
·
verified ·
1 Parent(s): b6fb1dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -12,7 +12,11 @@ from huggingface_hub import notebook_login
12
 
13
  from peft import LoraConfig, PeftConfig, PeftModel, get_peft_model, prepare_model_for_kbit_training
14
  from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
 
 
15
 
 
 
16
 
17
  bnb_config = BitsAndBytesConfig(
18
  load_in_4bit=True,
 
12
 
13
  from peft import LoraConfig, PeftConfig, PeftModel, get_peft_model, prepare_model_for_kbit_training
14
  from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
15
+ from huggingface_hub import login
16
+ import os
17
 
18
+ access_token = os.environ["HF_Token"]
19
+ login(token=access_token)
20
 
21
  bnb_config = BitsAndBytesConfig(
22
  load_in_4bit=True,