Update model.py
Browse files
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-
|
57 |
-
self.t5_tokenizer = T5Tokenizer.from_pretrained("t5-
|
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 = []
|