Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -55,9 +55,9 @@ def load_pipeline(model_name):
|
|
| 55 |
return PIPELINES[model_name]
|
| 56 |
repo = MODELS[model_name]["repo_id"]
|
| 57 |
if model_name == "secgpt-mini":
|
| 58 |
-
tokenizer = AutoTokenizer.from_pretrained(repo, trust_remote_code=True)
|
| 59 |
-
else:
|
| 60 |
tokenizer = AutoTokenizer.from_pretrained(repo, trust_remote_code=True, subfolder="models")
|
|
|
|
|
|
|
| 61 |
for dtype in (torch.bfloat16, torch.float16, torch.float32):
|
| 62 |
try:
|
| 63 |
if model_name == "secgpt-mini":
|
|
|
|
| 55 |
return PIPELINES[model_name]
|
| 56 |
repo = MODELS[model_name]["repo_id"]
|
| 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":
|