Den4ikAI
/

FRED-T5-XL-chitchat

Text2Text Generation

Inference Endpoints

text-generation-inference

Model card Files Files and versions Community

TeraSpace commited on Jun 3, 2023

Commit

a92b56a

•

1 Parent(s): cf5775b

ты бобо а ссылку кто напишет а???

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -18,8 +18,8 @@ use_cuda = torch.cuda.is_available()
 device = torch.device("cuda" if use_cuda else "cpu")
-t5_tokenizer = transformers.GPT2Tokenizer.from_pretrained(model_dir)
-t5_model = transformers.T5ForConditionalGeneration.from_pretrained(model_dir)
 while True:
     print('-'*80)

 device = torch.device("cuda" if use_cuda else "cpu")
+t5_tokenizer = transformers.GPT2Tokenizer.from_pretrained("Den4ikAI/FRED-T5-XL-chitchat")
+t5_model = transformers.T5ForConditionalGeneration.from_pretrained("Den4ikAI/FRED-T5-XL-chitchat")
 while True:
     print('-'*80)