Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3# Load model
4tokenizer = AutoTokenizer.from_pretrained("terrytaylorbonn/431_smollmv3_CPLT29")
5model = AutoModelForCausalLM.from_pretrained("terrytaylorbonn/431_smollmv3_CPLT29")
6
7# Generate text
8messages = [{"role": "user", "content": "Your prompt here"}]
9text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
10inputs = tokenizer(text, return_tensors="pt")
11
12outputs = model.generate(**inputs, max_new_tokens=200)
13response = tokenizer.decode(outputs[0][len(inputs.input_ids[0]):], skip_special_tokens=True)
14print(response)