flax-community
/

dansk-gpt-wiki

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

birgermoell commited on Jul 13, 2021

Commit

d1850c7

•

1 Parent(s): 1d2d61e

Updated eval script

Files changed (1) hide show

evaluate.py +2 -2

evaluate.py CHANGED Viewed

@@ -1,7 +1,7 @@
 from transformers import GPT2Tokenizer, GPT2Model, FlaxGPT2LMHeadModel, GPT2LMHeadModel, pipeline, set_seed
-tokenizer = GPT2Tokenizer.from_pretrained("flax-community/swe-gpt-wiki")
-model = GPT2LMHeadModel.from_pretrained("flax-community/swe-gpt-wiki")
 generator = pipeline('text-generation', model=model, tokenizer=tokenizer)

 from transformers import GPT2Tokenizer, GPT2Model, FlaxGPT2LMHeadModel, GPT2LMHeadModel, pipeline, set_seed
+tokenizer = GPT2Tokenizer.from_pretrained("flax-community/dansk-gpt-wiki")
+model = GPT2LMHeadModel.from_pretrained("flax-community/dansk-gpt-wiki")
 generator = pipeline('text-generation', model=model, tokenizer=tokenizer)