Shining-Data commited on
Commit
c8f58d7
·
verified ·
1 Parent(s): 24fbba8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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":