Views
No views yet
1from transformers import (
2 AutoTokenizer,
3 pipeline
4)
5
6model_name = "BrokenSoul/llama-2-7b-miniguanaco"
7
8tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
9tokenizer.pad_token = tokenizer.eos_token
10tokenizer.padding_side = "right"
11
12prompt = "What is a large language model?"
13pipe = pipeline(task="text-generation", model=model_name, tokenizer=tokenizer, max_length=200)
14result = pipe(f"<s>[INST] {prompt} [/INST]")
15print(result[0]['generated_text'])