peerawatchomp
commited on
Commit
•
a9eff47
1
Parent(s):
1cf2574
Create README.md
Browse files
README.md
ADDED
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
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)
|