khaiphan29 commited on
Commit
f20afb2
·
1 Parent(s): 54afbf6

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. src/myNLI.py +2 -2
src/myNLI.py CHANGED
@@ -26,8 +26,8 @@ class FactChecker:
26
  self.envir = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
27
 
28
  # Load LLM
29
- self.tokenizer = AutoTokenizer.from_pretrained("MoritzLaurer/mDeBERTa-v3-base-mnli-xnli") # LOAD mDEBERTa TOKENIZER
30
- self.mDeBertaModel = AutoModel.from_pretrained(f"src/mDeBERTa (ft) V6/mDeBERTa-v3-base-mnli-xnli-{self.INPUT_TYPE}") # LOAD FINETUNED MODEL
31
  # Load classifier model
32
  self.checkpoints = torch.load(f"src/mDeBERTa (ft) V6/{self.INPUT_TYPE}.pt", map_location=self.envir)
33
 
 
26
  self.envir = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
27
 
28
  # Load LLM
29
+ self.tokenizer = AutoTokenizer.from_pretrained("MoritzLaurer/mDeBERTa-v3-base-mnli-xnli", use_auth_token=False) # LOAD mDEBERTa TOKENIZER
30
+ self.mDeBertaModel = AutoModel.from_pretrained(f"src/mDeBERTa (ft) V6/mDeBERTa-v3-base-mnli-xnli-{self.INPUT_TYPE}", use_auth_token=False) # LOAD FINETUNED MODEL
31
  # Load classifier model
32
  self.checkpoints = torch.load(f"src/mDeBERTa (ft) V6/{self.INPUT_TYPE}.pt", map_location=self.envir)
33