Views
No views yet
transformers.meta-llama/Llama-3.2-1B1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained("your-username/Llama-3.2-1B_LoRA")
4tokenizer = AutoTokenizer.from_pretrained("your-username/Llama-3.2-1B_LoRA")
5
6inputs = tokenizer("Tell me something interesting about space.", return_tensors="pt")
7outputs = model.generate(**inputs, max_new_tokens=50)
8print(tokenizer.decode(outputs[0], skip_special_tokens=True))