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":
|