跳到内容

类型约束

我们可以要求补全限制为有效的 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
  • 我们也提供 自定义类型