




from typing import TYPE_CHECKING

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


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

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

     UpperCAmelCase_    :					Union[str, Any]					      =						_LazyModule(__name__, globals()['''__file__'''], _import_structure, module_spec=__spec__)












'''simple docstring'''



from typing import TYPE_CHECKING

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


UpperCAmelCase_    :					Any					      =						{
    '''configuration_retribert''': ['''RETRIBERT_PRETRAINED_CONFIG_ARCHIVE_MAP''', '''RetriBertConfig'''],
    '''tokenization_retribert''': ['''RetriBertTokenizer'''],
}

try:
     if not is_tokenizers_available():
          raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
     pass
else:
     UpperCAmelCase_    :					str					      =						['''RetriBertTokenizerFast''']

try:
     if not is_torch_available():
          raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
     pass
else:
     UpperCAmelCase_    :					Optional[int]					      =						[
         '''RETRIBERT_PRETRAINED_MODEL_ARCHIVE_LIST''',
         '''RetriBertModel''',
         '''RetriBertPreTrainedModel''',
     ]


if TYPE_CHECKING:
     from .configuration_retribert import RETRIBERT_PRETRAINED_CONFIG_ARCHIVE_MAP, RetriBertConfig
     from .tokenization_retribert import RetriBertTokenizer

     try:
          if not is_tokenizers_available():
               raise OptionalDependencyNotAvailable()
     except OptionalDependencyNotAvailable:
          pass
     else:
          from .tokenization_retribert_fast import RetriBertTokenizerFast

     try:
          if not is_torch_available():
               raise OptionalDependencyNotAvailable()
     except OptionalDependencyNotAvailable:
          pass
     else:
          from .modeling_retribert import (
              RETRIBERT_PRETRAINED_MODEL_ARCHIVE_LIST,
              RetriBertModel,
              RetriBertPreTrainedModel,
          )

else:
     import sys

     UpperCAmelCase_    :					Any					      =						_LazyModule(__name__, globals()['''__file__'''], _import_structure, module_spec=__spec__)






