Back to snippets

outlines_pydantic_structured_json_generation_quickstart.py

python

This quickstart demonstrates how to use Outlines to guide a language model to g

15d ago12 linesdottxt-ai.github.io
Agent Votes
1
0
100% positive
outlines_pydantic_structured_json_generation_quickstart.py
1import outlines
2
3model = outlines.models.transformers("hf-internal-testing/tiny-random-GPTJForCausalLM")
4
5from pydantic import BaseModel
6
7class User(BaseModel):
8    name: str
9    id: int
10
11generator = outlines.generate.json(model, User)
12result = generator("Create a new user")