Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3model_name = "your_username/your_model_name"
4
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6model = AutoModelForCausalLM.from_pretrained(model_name).to("cuda" or "cpu")
7
8prompt = "The following is a comment by a Nouns DAO member on the proposal to gift 1000 glasses to kids in need:"
9
10inputs = tokenizer(prompt, return_tensors="pt")
11outputs = model.generate(**inputs, max_new_tokens=100, repetition_penalty=1.15, temperature=0.7, top_p=0.9)
12
13generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
14print(generated_text)