ai-forever
/

FRED-T5-1.7B

Text2Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Update README.md

#3

by tinctura - opened Apr 10, 2023

base: refs/heads/main

←

from: refs/pr/3

Discussion Files changed

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)