Spaces:
Running
Running
nicholasKluge
commited on
Commit
•
ced5451
1
Parent(s):
4d8cb31
Update app.py
Browse files
app.py
CHANGED
@@ -21,9 +21,9 @@ model.to("cpu")
|
|
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 = """
|
|
|
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 = """
|