Justin commited on
Commit
e1defaf
·
2 Parent(s): 7452378cd19821

Merge branch 'main' of https://huggingface.co/spaces/jaydatech/frb-chatbot

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -29,8 +29,8 @@ HF_TOKEN = os.getenv("HF_TOKEN") # Load from Render environment variable
29
 
30
  device = "cuda" if torch.cuda.is_available() else "cpu"
31
 
32
- tokenizer = AutoTokenizer.from_pretrained(REPO_NAME, token=HF_TOKEN)
33
- #tokenizer = LlamaTokenizer.from_pretrained(REPO_NAME, use_auth_token=HF_TOKEN)
34
 
35
  model = AutoModelForCausalLM.from_pretrained(
36
  REPO_NAME,
 
29
 
30
  device = "cuda" if torch.cuda.is_available() else "cpu"
31
 
32
+ #tokenizer = AutoTokenizer.from_pretrained(REPO_NAME, token=HF_TOKEN)
33
+ tokenizer = LlamaTokenizer.from_pretrained(REPO_NAME, use_auth_token=HF_TOKEN)
34
 
35
  model = AutoModelForCausalLM.from_pretrained(
36
  REPO_NAME,