ai-forever
/

FRED-T5-large

Text2Text Generation

Inference Endpoints

text-generation-inference

Model card Files Files and versions Community

ai-forever commited on Apr 13, 2023

Commit

5b4be94

•

1 Parent(s): 7dcfe38

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -31,7 +31,7 @@ Total training time was around 35 days on 160 V100 GPUs + 5 days on 80 A100.
 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)