nicholasKluge commited on
Commit
3307d67
1 Parent(s): 37e5e69

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -21,9 +21,9 @@ model.to(device)
21
  rewardModel.to(device)
22
  toxicityModel.to(device)
23
 
24
- tokenizer = AutoTokenizer.from_pretrained(model_id, use_auth_token=token)
25
- rewardTokenizer = AutoTokenizer.from_pretrained(rewardmodel_id, use_auth_token=token)
26
- toxiciyTokenizer = AutoTokenizer.from_pretrained(toxicitymodel_id, use_auth_token=token)
27
 
28
 
29
  intro = """
 
21
  rewardModel.to(device)
22
  toxicityModel.to(device)
23
 
24
+ tokenizer = AutoTokenizer.from_pretrained(model_id)
25
+ rewardTokenizer = AutoTokenizer.from_pretrained(rewardmodel_id)
26
+ toxiciyTokenizer = AutoTokenizer.from_pretrained(toxicitymodel_id)
27
 
28
 
29
  intro = """