peerawatchomp
commited on
Commit
•
fa0655e
1
Parent(s):
a9eff47
Update README.md
Browse files
README.md
CHANGED
@@ -1,4 +1,7 @@
|
|
1 |
MODEL_NAME = 'nonamenlp/news_gen'
|
|
|
2 |
TOKENIZER_NAME = "nonamenlp/news_gen"
|
|
|
3 |
trained_model = MT5ForConditionalGeneration.from_pretrained(MODEL_NAME, return_dict=True)
|
|
|
4 |
tokenizer = T5Tokenizer.from_pretrained(TOKENIZER_NAME)
|
1 |
MODEL_NAME = 'nonamenlp/news_gen'
|
2 |
+
|
3 |
TOKENIZER_NAME = "nonamenlp/news_gen"
|
4 |
+
|
5 |
trained_model = MT5ForConditionalGeneration.from_pretrained(MODEL_NAME, return_dict=True)
|
6 |
+
|
7 |
tokenizer = T5Tokenizer.from_pretrained(TOKENIZER_NAME)
|