JosephH commited on
Commit
fb8239c
1 Parent(s): 9cc1e73

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +2 -2
model.py CHANGED
@@ -53,8 +53,8 @@ class GPT2PPLV2:
53
  self.stride = 51
54
  self.threshold = 0.7
55
 
56
- self.t5_model = transformers.AutoModelForSeq2SeqLM.from_pretrained("t5-base").to(device)
57
- self.t5_tokenizer = T5Tokenizer.from_pretrained("t5-larger", model_max_length=512)
58
 
59
  def apply_extracted_fills(self, masked_texts, extracted_fills):
60
  texts = []
 
53
  self.stride = 51
54
  self.threshold = 0.7
55
 
56
+ self.t5_model = transformers.AutoModelForSeq2SeqLM.from_pretrained("t5-large").to(device)
57
+ self.t5_tokenizer = T5Tokenizer.from_pretrained("t5-large", model_max_length=512)
58
 
59
  def apply_extracted_fills(self, masked_texts, extracted_fills):
60
  texts = []