ai-forever
/

FRED-T5-1.7B

Text2Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

tinctura commited on Apr 10, 2023

Commit

d561803

•

1 Parent(s): 75e6b4e

Update README.md

minor changes) removed extra parenthesis

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)