o پiW@sddlmZddlZddlmZddlmZmZddlm Z ddl m Z m Z ee fdej jde d efd dZ e e e fdej jde fd d ZdS) )singledispatchN) RegexGuide)SequenceGeneratorAdapterVisionSequenceGeneratorAdapter)TransformersVision)Sampler multinomialfsmsamplerreturncC4ddlm}t||j}||j|d}t|||SNr)GuideLogitsProcessor) tokenizerguide)outlines.processorsrrfrom_interegular_fsmrrmodelr r rrlogits_processorrI/home/ubuntu/.local/lib/python3.10/site-packages/outlines/generate/fsm.pyr s  cCr r )rrrrrrrrrr fsm_visions  r) functoolsr interegularoutlines.fsm.guideroutlines.generate.apirroutlines.modelsroutlines.samplersrrr FSMregisterrrrrrs"     "