o پi@sddlmZddlmZmZddlmZmZddlm Z m Z ee fde defddZ e ee fde fd d Z e ee fd ede defd d ZdS))singledispatch)SequenceGeneratorAdapterVisionSequenceGeneratorAdapter)OpenAITransformersVision)Sampler multinomialsamplerreturncCs t|d|S)aGenerate text with a `Transformer` model. Note ---- Python 3.11 allows dispatching on Union types and this should greatly simplify the code. Arguments --------- model: An instance of `Transformer` that represents a model from the `transformers` library. sampler: The sampling algorithm to use to generate token ids from the logits distribution. Returns ------- A `SequenceGeneratorAdapter` instance that generates text. N)rmodelr r J/home/ubuntu/.local/lib/python3.10/site-packages/outlines/generate/text.pytext s rcCs t|d|S)N)rr r r r text_vision%s rr cCst|ts td|S)NzZThe OpenAI API does not support any other sampling algorithm than the multinomial sampler.) isinstancerNotImplementedErrorr r r r text_openai*s rN) functoolsroutlines.generate.apirroutlines.modelsrroutlines.samplersrrrregisterrrr r r rs "