Views
No views yet

1# Use a pipeline as a high-level helper
2from transformers import pipeline
3
4messages = [
5 {"role": "user", "content": "Who are you?"},
6]
7pipe = pipeline("text-generation", model="Spestly/Athena-1-7B")
8pipe(messages)
9
10# Load model directly
11from transformers import AutoTokenizer, AutoModelForCausalLM
12
13tokenizer = AutoTokenizer.from_pretrained("Spestly/Athena-1-7B")
14model = AutoModelForCausalLM.from_pretrained("Spestly/Athena-1-7B")