类型约束
我们可以要求补全限制为有效的 Python 类型
from outlines import models, generate
model = models.transformers("microsoft/Phi-3-mini-4k-instruct")
generator = generate.format(model, int)
answer = generator("When I was 6 my sister was half my age. Now I’m 70 how old is my sister?")
print(answer)
# 67
目前支持以下类型
- int
- float
- bool
- datetime.date
- datetime.time
- datetime.datetime
- 我们也提供 自定义类型