


from typing import TYPE_CHECKING

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


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

try:
 if not is_torch_available():
  raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
 pass
else:
 lowerCamelCase						 =	[
     'Constraint',
     'ConstraintListState',
     'DisjunctiveConstraint',
     'PhrasalConstraint',
 ]
 lowerCamelCase						 =	[
     'BeamHypotheses',
     'BeamScorer',
     'BeamSearchScorer',
     'ConstrainedBeamSearchScorer',
 ]
 lowerCamelCase						 =	[
     'EpsilonLogitsWarper',
     'EtaLogitsWarper',
     'ForcedBOSTokenLogitsProcessor',
     'ForcedEOSTokenLogitsProcessor',
     'HammingDiversityLogitsProcessor',
     'InfNanRemoveLogitsProcessor',
     'LogitsProcessor',
     'LogitsProcessorList',
     'LogitsWarper',
     'MinLengthLogitsProcessor',
     'MinNewTokensLengthLogitsProcessor',
     'NoBadWordsLogitsProcessor',
     'NoRepeatNGramLogitsProcessor',
     'PrefixConstrainedLogitsProcessor',
     'RepetitionPenaltyLogitsProcessor',
     'SequenceBiasLogitsProcessor',
     'EncoderRepetitionPenaltyLogitsProcessor',
     'TemperatureLogitsWarper',
     'TopKLogitsWarper',
     'TopPLogitsWarper',
     'TypicalLogitsWarper',
     'EncoderNoRepeatNGramLogitsProcessor',
     'ExponentialDecayLengthPenalty',
     'LogitNormalization',
 ]
 lowerCamelCase						 =	[
     'MaxNewTokensCriteria',
     'MaxLengthCriteria',
     'MaxTimeCriteria',
     'StoppingCriteria',
     'StoppingCriteriaList',
     'validate_stopping_criteria',
 ]
 lowerCamelCase						 =	[
     '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:
 lowerCamelCase						 =	[
     'TFForcedBOSTokenLogitsProcessor',
     'TFForcedEOSTokenLogitsProcessor',
     'TFLogitsProcessor',
     'TFLogitsProcessorList',
     'TFLogitsWarper',
     'TFMinLengthLogitsProcessor',
     'TFNoBadWordsLogitsProcessor',
     'TFNoRepeatNGramLogitsProcessor',
     'TFRepetitionPenaltyLogitsProcessor',
     'TFTemperatureLogitsWarper',
     'TFTopKLogitsWarper',
     'TFTopPLogitsWarper',
     'TFForceTokensLogitsProcessor',
     'TFSuppressTokensAtBeginLogitsProcessor',
     'TFSuppressTokensLogitsProcessor',
 ]
 lowerCamelCase						 =	[
     '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:
 lowerCamelCase						 =	[
     'FlaxForcedBOSTokenLogitsProcessor',
     'FlaxForcedEOSTokenLogitsProcessor',
     'FlaxLogitsProcessor',
     'FlaxLogitsProcessorList',
     'FlaxLogitsWarper',
     'FlaxMinLengthLogitsProcessor',
     'FlaxTemperatureLogitsWarper',
     'FlaxTopKLogitsWarper',
     'FlaxTopPLogitsWarper',
 ]
 lowerCamelCase						 =	[
     '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

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






from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available


lowerCamelCase						 =	{'configuration_visual_bert': ['VISUAL_BERT_PRETRAINED_CONFIG_ARCHIVE_MAP', 'VisualBertConfig']}

try:
 if not is_torch_available():
  raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
 pass
else:
 lowerCamelCase						 =	[
     'VISUAL_BERT_PRETRAINED_MODEL_ARCHIVE_LIST',
     'VisualBertForMultipleChoice',
     'VisualBertForPreTraining',
     'VisualBertForQuestionAnswering',
     'VisualBertForRegionToPhraseAlignment',
     'VisualBertForVisualReasoning',
     'VisualBertLayer',
     'VisualBertModel',
     'VisualBertPreTrainedModel',
 ]


if TYPE_CHECKING:
 from .configuration_visual_bert import VISUAL_BERT_PRETRAINED_CONFIG_ARCHIVE_MAP, VisualBertConfig

 try:
  if not is_torch_available():
   raise OptionalDependencyNotAvailable()
 except OptionalDependencyNotAvailable:
  pass
 else:
  from .modeling_visual_bert import (
      VISUAL_BERT_PRETRAINED_MODEL_ARCHIVE_LIST,
      VisualBertForMultipleChoice,
      VisualBertForPreTraining,
      VisualBertForQuestionAnswering,
      VisualBertForRegionToPhraseAlignment,
      VisualBertForVisualReasoning,
      VisualBertLayer,
      VisualBertModel,
      VisualBertPreTrainedModel,
  )


else:
 import sys

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


