Pedrinho-Dev01 commited on
Commit
56417b0
·
1 Parent(s): eb496e4
Files changed (1) hide show
  1. api.py +1 -1
api.py CHANGED
@@ -50,7 +50,7 @@ app.add_middleware(
50
  class ModelBundle:
51
  def __init__(self, repo_id: str, model_class):
52
  print(f"Loading {repo_id} …")
53
- self.tokenizer = AutoTokenizer.from_pretrained(repo_id, use_fast=False)
54
  self.model = model_class.from_pretrained(repo_id)
55
  self.model.to(DEVICE)
56
  self.model.eval()
 
50
  class ModelBundle:
51
  def __init__(self, repo_id: str, model_class):
52
  print(f"Loading {repo_id} …")
53
+ self.tokenizer = AutoTokenizer.from_pretrained(repo_id)
54
  self.model = model_class.from_pretrained(repo_id)
55
  self.model.to(DEVICE)
56
  self.model.eval()