Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -57,7 +57,7 @@ def load_pipeline(model_name):
|
|
| 57 |
if model_name == "secgpt-mini":
|
| 58 |
tokenizer = AutoTokenizer.from_pretrained(repo, trust_remote_code=True, subfolder="models")
|
| 59 |
else:
|
| 60 |
-
tokenizer = AutoTokenizer.from_pretrained(repo, trust_remote_code=True
|
| 61 |
for dtype in (torch.bfloat16, torch.float16, torch.float32):
|
| 62 |
try:
|
| 63 |
if model_name == "secgpt-mini":
|
|
|
|
| 57 |
if model_name == "secgpt-mini":
|
| 58 |
tokenizer = AutoTokenizer.from_pretrained(repo, trust_remote_code=True, subfolder="models")
|
| 59 |
else:
|
| 60 |
+
tokenizer = AutoTokenizer.from_pretrained(repo, trust_remote_code=True)
|
| 61 |
for dtype in (torch.bfloat16, torch.float16, torch.float32):
|
| 62 |
try:
|
| 63 |
if model_name == "secgpt-mini":
|