Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained(
4 "your-username/model-name",
5 torch_dtype=torch.float16,
6 device_map="auto"
7)
8tokenizer = AutoTokenizer.from_pretrained("your-username/model-name")
9
10# Generate text
11prompt = "What is pet insurance?"
12inputs = tokenizer(prompt, return_tensors="pt")
13outputs = model.generate(**inputs, max_length=100)
14response = tokenizer.decode(outputs[0], skip_special_tokens=True)1@misc{lassie-experiment-outlandish,
2 title={LLaMA-2-7B Fine-tuned with Outlandish Lassie Data},
3 author={Outland Colossus Team},
4 year={2024},
5 url={https://huggingface.co/your-username/model-name}
6}