TiberiuCristianLeon commited on
Commit
544f6db
·
verified ·
1 Parent(s): 79c9157

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -65,7 +65,7 @@ class Translators:
65
  def mitre(self):
66
  from transformers import AutoModel, AutoTokenizer
67
  tokenizer = AutoTokenizer.from_pretrained(self.model_name, trust_remote_code=True, use_fast=False)
68
- model = AutoModel.from_pretrained(self.model_name, trust_remote_code=True, use_fast=False).to(self.device)
69
  # model = AutoModel.from_pretrained(self.model_name, trust_remote_code=True)
70
  # model.half() # recommended for GPU
71
  model.eval()
 
65
  def mitre(self):
66
  from transformers import AutoModel, AutoTokenizer
67
  tokenizer = AutoTokenizer.from_pretrained(self.model_name, trust_remote_code=True, use_fast=False)
68
+ model = AutoModel.from_pretrained(self.model_name, trust_remote_code=True).to(self.device)
69
  # model = AutoModel.from_pretrained(self.model_name, trust_remote_code=True)
70
  # model.half() # recommended for GPU
71
  model.eval()