ai-forever
/

FRED-T5-1.7B

text2text-generation

text-generation-inference

Model card Files Files and versions

ai-forever commited on Apr 13, 2023

Commit

4b8b16d

·

1 Parent(s): 9d515ce

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -32,7 +32,7 @@ Total training time was around 45 days on 112 A100 GPUs.
 import torch
 from transformers import GPT2Tokenizer, T5ForConditionalGeneration
 tokenizer = GPT2Tokenizer.from_pretrained('ai-forever/FRED-T5-1.7B',eos_token='</s>')
-model = T5ForConditionalGeneration.from_pretrained(('ai-forever/FRED-T5-1.7B')
 device='cuda'
 model.to(device)

 import torch
 from transformers import GPT2Tokenizer, T5ForConditionalGeneration
 tokenizer = GPT2Tokenizer.from_pretrained('ai-forever/FRED-T5-1.7B',eos_token='</s>')
+model = T5ForConditionalGeneration.from_pretrained('ai-forever/FRED-T5-1.7B')
 device='cuda'
 model.to(device)