Views
No views yet
1from transformers import pipeline
2from transformers import GPT2Model, GPT2TokenizerFast, AutoModelForCausalLM, AutoTokenizer
3
4model_name = "tupleblog/generate-thai-lyrics"
5model = AutoModelForCausalLM.from_pretrained(model_name)
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7model.config.pad_token_id = model.config.eos_token_id
8
9
10nlp = pipeline(
11 "text-generation",
12 model=model,
13 tokenizer=tokenizer
14)
15text = "ความรัก"
16nlp(text, max_length=100, top_k=40, temperature=0.8) # varying the temperature and top-k produce different output