Justin commited on
Commit
6d5d500
·
1 Parent(s): 5b6a521

Removed requirements ... more pt 7

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,