Commit
•
4b8b16d
1
Parent(s):
9d515ce
Update README.md
Browse files
README.md
CHANGED
@@ -32,7 +32,7 @@ Total training time was around 45 days on 112 A100 GPUs.
|
|
32 |
import torch
|
33 |
from transformers import GPT2Tokenizer, T5ForConditionalGeneration
|
34 |
tokenizer = GPT2Tokenizer.from_pretrained('ai-forever/FRED-T5-1.7B',eos_token='</s>')
|
35 |
-
model = T5ForConditionalGeneration.from_pretrained(
|
36 |
device='cuda'
|
37 |
model.to(device)
|
38 |
|
|
|
32 |
import torch
|
33 |
from transformers import GPT2Tokenizer, T5ForConditionalGeneration
|
34 |
tokenizer = GPT2Tokenizer.from_pretrained('ai-forever/FRED-T5-1.7B',eos_token='</s>')
|
35 |
+
model = T5ForConditionalGeneration.from_pretrained('ai-forever/FRED-T5-1.7B')
|
36 |
device='cuda'
|
37 |
model.to(device)
|
38 |
|