jaydatech commited on
Commit
cd19821
·
verified ·
1 Parent(s): 6d5d500

Update app.py

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