
"""simple docstring"""


from typing import TYPE_CHECKING

from ..utils import OptionalDependencyNotAvailable, _LazyModule, is_flax_available, is_tf_available, is_torch_available


_lowercase											=			{
    '''configuration_utils''': ['''GenerationConfig'''],
    '''streamers''': ['''TextIteratorStreamer''', '''TextStreamer'''],
}

try:
		if not is_torch_available():
				raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
		pass
else:
		_lowercase											=			[
		    '''Constraint''',
		    '''ConstraintListState''',
		    '''DisjunctiveConstraint''',
		    '''PhrasalConstraint''',
		]
		_lowercase											=			[
		    '''BeamHypotheses''',
		    '''BeamScorer''',
		    '''BeamSearchScorer''',
		    '''ConstrainedBeamSearchScorer''',
		]
		_lowercase											=			[
		    '''EpsilonLogitsWarper''',
		    '''EtaLogitsWarper''',
		    '''ForcedBOSTokenLogitsProcessor''',
		    '''ForcedEOSTokenLogitsProcessor''',
		    '''HammingDiversityLogitsProcessor''',
		    '''InfNanRemoveLogitsProcessor''',
		    '''LogitsProcessor''',
		    '''LogitsProcessorList''',
		    '''LogitsWarper''',
		    '''MinLengthLogitsProcessor''',
		    '''MinNewTokensLengthLogitsProcessor''',
		    '''NoBadWordsLogitsProcessor''',
		    '''NoRepeatNGramLogitsProcessor''',
		    '''PrefixConstrainedLogitsProcessor''',
		    '''RepetitionPenaltyLogitsProcessor''',
		    '''SequenceBiasLogitsProcessor''',
		    '''EncoderRepetitionPenaltyLogitsProcessor''',
		    '''TemperatureLogitsWarper''',
		    '''TopKLogitsWarper''',
		    '''TopPLogitsWarper''',
		    '''TypicalLogitsWarper''',
		    '''EncoderNoRepeatNGramLogitsProcessor''',
		    '''ExponentialDecayLengthPenalty''',
		    '''LogitNormalization''',
		]
		_lowercase											=			[
		    '''MaxNewTokensCriteria''',
		    '''MaxLengthCriteria''',
		    '''MaxTimeCriteria''',
		    '''StoppingCriteria''',
		    '''StoppingCriteriaList''',
		    '''validate_stopping_criteria''',
		]
		_lowercase											=			[
		    '''GenerationMixin''',
		    '''top_k_top_p_filtering''',
		    '''GreedySearchEncoderDecoderOutput''',
		    '''GreedySearchDecoderOnlyOutput''',
		    '''SampleEncoderDecoderOutput''',
		    '''SampleDecoderOnlyOutput''',
		    '''BeamSearchEncoderDecoderOutput''',
		    '''BeamSearchDecoderOnlyOutput''',
		    '''BeamSampleEncoderDecoderOutput''',
		    '''BeamSampleDecoderOnlyOutput''',
		    '''ContrastiveSearchEncoderDecoderOutput''',
		    '''ContrastiveSearchDecoderOnlyOutput''',
		]

try:
		if not is_tf_available():
				raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
		pass
else:
		_lowercase											=			[
		    '''TFForcedBOSTokenLogitsProcessor''',
		    '''TFForcedEOSTokenLogitsProcessor''',
		    '''TFLogitsProcessor''',
		    '''TFLogitsProcessorList''',
		    '''TFLogitsWarper''',
		    '''TFMinLengthLogitsProcessor''',
		    '''TFNoBadWordsLogitsProcessor''',
		    '''TFNoRepeatNGramLogitsProcessor''',
		    '''TFRepetitionPenaltyLogitsProcessor''',
		    '''TFTemperatureLogitsWarper''',
		    '''TFTopKLogitsWarper''',
		    '''TFTopPLogitsWarper''',
		    '''TFForceTokensLogitsProcessor''',
		    '''TFSuppressTokensAtBeginLogitsProcessor''',
		    '''TFSuppressTokensLogitsProcessor''',
		]
		_lowercase											=			[
		    '''TFGenerationMixin''',
		    '''tf_top_k_top_p_filtering''',
		    '''TFGreedySearchDecoderOnlyOutput''',
		    '''TFGreedySearchEncoderDecoderOutput''',
		    '''TFSampleEncoderDecoderOutput''',
		    '''TFSampleDecoderOnlyOutput''',
		    '''TFBeamSearchEncoderDecoderOutput''',
		    '''TFBeamSearchDecoderOnlyOutput''',
		    '''TFBeamSampleEncoderDecoderOutput''',
		    '''TFBeamSampleDecoderOnlyOutput''',
		    '''TFContrastiveSearchEncoderDecoderOutput''',
		    '''TFContrastiveSearchDecoderOnlyOutput''',
		]

try:
		if not is_flax_available():
				raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
		pass
else:
		_lowercase											=			[
		    '''FlaxForcedBOSTokenLogitsProcessor''',
		    '''FlaxForcedEOSTokenLogitsProcessor''',
		    '''FlaxLogitsProcessor''',
		    '''FlaxLogitsProcessorList''',
		    '''FlaxLogitsWarper''',
		    '''FlaxMinLengthLogitsProcessor''',
		    '''FlaxTemperatureLogitsWarper''',
		    '''FlaxTopKLogitsWarper''',
		    '''FlaxTopPLogitsWarper''',
		]
		_lowercase											=			[
		    '''FlaxGenerationMixin''',
		    '''FlaxGreedySearchOutput''',
		    '''FlaxSampleOutput''',
		    '''FlaxBeamSearchOutput''',
		]

if TYPE_CHECKING:
		from .configuration_utils import GenerationConfig
		from .streamers import TextIteratorStreamer, TextStreamer

		try:
				if not is_torch_available():
						raise OptionalDependencyNotAvailable()
		except OptionalDependencyNotAvailable:
				pass
		else:
				from .beam_constraints import Constraint, ConstraintListState, DisjunctiveConstraint, PhrasalConstraint
				from .beam_search import BeamHypotheses, BeamScorer, BeamSearchScorer, ConstrainedBeamSearchScorer
				from .logits_process import (
				    EncoderNoRepeatNGramLogitsProcessor,
				    EncoderRepetitionPenaltyLogitsProcessor,
				    EpsilonLogitsWarper,
				    EtaLogitsWarper,
				    ExponentialDecayLengthPenalty,
				    ForcedBOSTokenLogitsProcessor,
				    ForcedEOSTokenLogitsProcessor,
				    HammingDiversityLogitsProcessor,
				    InfNanRemoveLogitsProcessor,
				    LogitNormalization,
				    LogitsProcessor,
				    LogitsProcessorList,
				    LogitsWarper,
				    MinLengthLogitsProcessor,
				    MinNewTokensLengthLogitsProcessor,
				    NoBadWordsLogitsProcessor,
				    NoRepeatNGramLogitsProcessor,
				    PrefixConstrainedLogitsProcessor,
				    RepetitionPenaltyLogitsProcessor,
				    SequenceBiasLogitsProcessor,
				    TemperatureLogitsWarper,
				    TopKLogitsWarper,
				    TopPLogitsWarper,
				    TypicalLogitsWarper,
				)
				from .stopping_criteria import (
				    MaxLengthCriteria,
				    MaxNewTokensCriteria,
				    MaxTimeCriteria,
				    StoppingCriteria,
				    StoppingCriteriaList,
				    validate_stopping_criteria,
				)
				from .utils import (
				    BeamSampleDecoderOnlyOutput,
				    BeamSampleEncoderDecoderOutput,
				    BeamSearchDecoderOnlyOutput,
				    BeamSearchEncoderDecoderOutput,
				    ContrastiveSearchDecoderOnlyOutput,
				    ContrastiveSearchEncoderDecoderOutput,
				    GenerationMixin,
				    GreedySearchDecoderOnlyOutput,
				    GreedySearchEncoderDecoderOutput,
				    SampleDecoderOnlyOutput,
				    SampleEncoderDecoderOutput,
				    top_k_top_p_filtering,
				)

		try:
				if not is_tf_available():
						raise OptionalDependencyNotAvailable()
		except OptionalDependencyNotAvailable:
				pass
		else:
				from .tf_logits_process import (
				    TFForcedBOSTokenLogitsProcessor,
				    TFForcedEOSTokenLogitsProcessor,
				    TFForceTokensLogitsProcessor,
				    TFLogitsProcessor,
				    TFLogitsProcessorList,
				    TFLogitsWarper,
				    TFMinLengthLogitsProcessor,
				    TFNoBadWordsLogitsProcessor,
				    TFNoRepeatNGramLogitsProcessor,
				    TFRepetitionPenaltyLogitsProcessor,
				    TFSuppressTokensAtBeginLogitsProcessor,
				    TFSuppressTokensLogitsProcessor,
				    TFTemperatureLogitsWarper,
				    TFTopKLogitsWarper,
				    TFTopPLogitsWarper,
				)
				from .tf_utils import (
				    TFBeamSampleDecoderOnlyOutput,
				    TFBeamSampleEncoderDecoderOutput,
				    TFBeamSearchDecoderOnlyOutput,
				    TFBeamSearchEncoderDecoderOutput,
				    TFContrastiveSearchDecoderOnlyOutput,
				    TFContrastiveSearchEncoderDecoderOutput,
				    TFGenerationMixin,
				    TFGreedySearchDecoderOnlyOutput,
				    TFGreedySearchEncoderDecoderOutput,
				    TFSampleDecoderOnlyOutput,
				    TFSampleEncoderDecoderOutput,
				    tf_top_k_top_p_filtering,
				)

		try:
				if not is_flax_available():
						raise OptionalDependencyNotAvailable()
		except OptionalDependencyNotAvailable:
				pass
		else:
				from .flax_logits_process import (
				    FlaxForcedBOSTokenLogitsProcessor,
				    FlaxForcedEOSTokenLogitsProcessor,
				    FlaxLogitsProcessor,
				    FlaxLogitsProcessorList,
				    FlaxLogitsWarper,
				    FlaxMinLengthLogitsProcessor,
				    FlaxTemperatureLogitsWarper,
				    FlaxTopKLogitsWarper,
				    FlaxTopPLogitsWarper,
				)
				from .flax_utils import FlaxBeamSearchOutput, FlaxGenerationMixin, FlaxGreedySearchOutput, FlaxSampleOutput
else:
		import sys

		_lowercase											=			_LazyModule(__name__, globals()['''__file__'''], _import_structure, module_spec=__spec__)

"""simple docstring"""


from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available, is_vision_available


_lowercase											=			{'''configuration_vilt''': ['''VILT_PRETRAINED_CONFIG_ARCHIVE_MAP''', '''ViltConfig''']}

try:
		if not is_vision_available():
				raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
		pass
else:
		_lowercase											=			['''ViltFeatureExtractor''']
		_lowercase											=			['''ViltImageProcessor''']
		_lowercase											=			['''ViltProcessor''']

try:
		if not is_torch_available():
				raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
		pass
else:
		_lowercase											=			[
		    '''VILT_PRETRAINED_MODEL_ARCHIVE_LIST''',
		    '''ViltForImageAndTextRetrieval''',
		    '''ViltForImagesAndTextClassification''',
		    '''ViltForTokenClassification''',
		    '''ViltForMaskedLM''',
		    '''ViltForQuestionAnswering''',
		    '''ViltLayer''',
		    '''ViltModel''',
		    '''ViltPreTrainedModel''',
		]


if TYPE_CHECKING:
		from .configuration_vilt import VILT_PRETRAINED_CONFIG_ARCHIVE_MAP, ViltConfig

		try:
				if not is_vision_available():
						raise OptionalDependencyNotAvailable()
		except OptionalDependencyNotAvailable:
				pass
		else:
				from .feature_extraction_vilt import ViltFeatureExtractor
				from .image_processing_vilt import ViltImageProcessor
				from .processing_vilt import ViltProcessor

		try:
				if not is_torch_available():
						raise OptionalDependencyNotAvailable()
		except OptionalDependencyNotAvailable:
				pass
		else:
				from .modeling_vilt import (
				    VILT_PRETRAINED_MODEL_ARCHIVE_LIST,
				    ViltForImageAndTextRetrieval,
				    ViltForImagesAndTextClassification,
				    ViltForMaskedLM,
				    ViltForQuestionAnswering,
				    ViltForTokenClassification,
				    ViltLayer,
				    ViltModel,
				    ViltPreTrainedModel,
				)


else:
		import sys

		_lowercase											=			_LazyModule(__name__, globals()['''__file__'''], _import_structure)