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-1.5B") # Update model name
8print(pipe(messages))
9
10# Load model directly
11from transformers import AutoTokenizer, AutoModelForCausalLM
12
13tokenizer = AutoTokenizer.from_pretrained("Spestly/Athena-1-1.5B") # Update model name
14model = AutoModelForCausalLM.from_pretrained("Spestly/Athena-1-1.5B") # Update model name