File size: 254 Bytes
c37a02b
 
5de9a55
c37a02b
 
 
782bba2
c37a02b
1
2
3
4
5
6
7
8
# Use a pipeline as a high-level helper
from transformers import pipeline

messages = [
    {"role": "user", "content": "Who are you?"},
]
pipe = pipeline("text-generation", legacy=false, model="Mxode/minicoder-7M-init",max_new_tokens=100)
pipe(messages)