Update README.md
Browse files
README.md
CHANGED
@@ -44,12 +44,9 @@ Max Sequence Length: 512
|
|
44 |
|
45 |
Example usage:
|
46 |
|
47 |
-
base_model_name = 't5-large'
|
48 |
-
|
49 |
-
|
50 |
-
|
51 |
-
model = T5ForConditionalGeneration.from_pretrained('trichter/t5-DistillingSbS-ABSA')
|
52 |
-
|
53 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
54 |
|
55 |
model.to(device)
|
|
|
44 |
|
45 |
Example usage:
|
46 |
|
47 |
+
base_model_name = 't5-large'
|
48 |
+
tokenizer = T5Tokenizer.from_pretrained(base_model_name, model_max_length=512)
|
49 |
+
model = T5ForConditionalGeneration.from_pretrained('trichter/t5-DistillingSbS-ABSA')
|
|
|
|
|
|
|
50 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
51 |
|
52 |
model.to(device)
|