Spaces:
Runtime error
Runtime error
Commit
β’
7735bc8
1
Parent(s):
a8207d3
revert
Browse files
app.py
CHANGED
@@ -348,12 +348,12 @@ def recreate_bm25_idx(
|
|
348 |
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
|
349 |
|
350 |
safety_tokenizer = tokenizer = AutoTokenizer.from_pretrained(
|
351 |
-
"
|
352 |
use_auth_token=HF_TOKEN,
|
353 |
)
|
354 |
safety_model = model = (
|
355 |
AutoModelForSeq2SeqLM.from_pretrained(
|
356 |
-
"
|
357 |
use_auth_token=HF_TOKEN,
|
358 |
)
|
359 |
.half()
|
|
|
348 |
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
|
349 |
|
350 |
safety_tokenizer = tokenizer = AutoTokenizer.from_pretrained(
|
351 |
+
"salexashenko/T5-Base-ROT-epoch-2-train-loss-1.3495-val-loss-1.4164",
|
352 |
use_auth_token=HF_TOKEN,
|
353 |
)
|
354 |
safety_model = model = (
|
355 |
AutoModelForSeq2SeqLM.from_pretrained(
|
356 |
+
"salexashenko/T5-Base-ROT-epoch-2-train-loss-1.3495-val-loss-1.4164",
|
357 |
use_auth_token=HF_TOKEN,
|
358 |
)
|
359 |
.half()
|