jaydatech commited on
Commit
bbcace9
·
verified ·
1 Parent(s): e1defaf

Change tokenizer

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,