Skip to content

Commit

Permalink
Add OpenAI to LogitsGenerator
Browse files Browse the repository at this point in the history
  • Loading branch information
@ulfaslak @rlouf
ulfaslak authored and rlouf committed Nov 27, 2024
1 parent 2db34c5 commit 66f6627
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion outlines/models/__init__.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -16,4 +16,4 @@
from .transformers_vision import TransformersVision, transformers_vision
from .vllm import VLLM, vllm

LogitsGenerator = Union[Transformers, LlamaCpp, ExLlamaV2Model, MLXLM, VLLM]
LogitsGenerator = Union[Transformers, LlamaCpp, OpenAI, ExLlamaV2Model, MLXLM, VLLM]

0 comments on commit 66f6627

Please sign in to comment.