truongpdd
/

vietnews-gpt2

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

truongpdd commited on Sep 21, 2022

Commit

71232c5

•

1 Parent(s): 511170e

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -4,6 +4,6 @@
 from transformers import AutoModelForCausalLM, AutoTokenizer
-tokenizer = AutoTokenizer.from_pretrained('truongpdd/vietnamese-news-gpt2-5epochs')
-model = AutoModelForCausalLM.from_pretrained('truongpdd/vietnamese-news-gpt2-5epochs', from_flax=True)
 ```

 from transformers import AutoModelForCausalLM, AutoTokenizer
+tokenizer = AutoTokenizer.from_pretrained('truongpdd/vietnews-gpt2')
+model = AutoModelForCausalLM.from_pretrained('truongpdd/vietnews-gpt2')
 ```