Den4ikAI
/

FRED-T5-XL-chitchat

Text2Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

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

#2

by TeraSpace - opened Jun 3, 2023

base: refs/heads/main

←

from: refs/pr/2

Discussion Files changed

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)