harpreetsahota commited on
Commit
7e7769f
1 Parent(s): 6ed0c17

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ tokenizer_fim.add_special_tokens({
34
  "pad_token": EOD,
35
  })
36
 
37
- tokenizer = AutoTokenizer.from_pretrained("bigcode/starcoder", use_auth_token=token, force_download=True)
38
 
39
  model = AutoModelForCausalLM.from_pretrained("Deci/DeciCoder-1b", trust_remote_code=True, use_auth_token=token, force_download=True).to(device)
40
 
 
34
  "pad_token": EOD,
35
  })
36
 
37
+ tokenizer = AutoTokenizer.from_pretrained("Deci/DeciCoder-1b", use_auth_token=token, force_download=True)
38
 
39
  model = AutoModelForCausalLM.from_pretrained("Deci/DeciCoder-1b", trust_remote_code=True, use_auth_token=token, force_download=True).to(device)
40