Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3tokenizer = AutoTokenizer.from_pretrained("distilgpt2")
4model = AutoModelForCausalLM.from_pretrained("YOUR_HF_NAME/platz109M")
5
6prompt = "Leg workouts are very important"
7inputs = tokenizer(prompt, return_tensors="pt")
8out = model.generate(**inputs, max_new_tokens=50, do_sample=True, top_p=0.92)
9print(tokenizer.decode(out[0], skip_special_tokens=True))