


'''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_swin2sr": ["SWIN2SR_PRETRAINED_CONFIG_ARCHIVE_MAP", "Swin2SRConfig"],
}


try:
   if not is_torch_available():
      raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
   pass
else:
   lowercase__					  =		[
       "SWIN2SR_PRETRAINED_MODEL_ARCHIVE_LIST",
       "Swin2SRForImageSuperResolution",
       "Swin2SRModel",
       "Swin2SRPreTrainedModel",
   ]


try:
   if not is_vision_available():
      raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
   pass
else:
   lowercase__					  =		["Swin2SRImageProcessor"]


if TYPE_CHECKING:
   from .configuration_swinasr import SWIN2SR_PRETRAINED_CONFIG_ARCHIVE_MAP, SwinaSRConfig

   try:
      if not is_torch_available():
         raise OptionalDependencyNotAvailable()
   except OptionalDependencyNotAvailable:
      pass
   else:
      from .modeling_swinasr import (
          SWIN2SR_PRETRAINED_MODEL_ARCHIVE_LIST,
          SwinaSRForImageSuperResolution,
          SwinaSRModel,
          SwinaSRPreTrainedModel,
      )

   try:
      if not is_vision_available():
         raise OptionalDependencyNotAvailable()
   except OptionalDependencyNotAvailable:
      pass
   else:
      from .image_processing_swinasr import SwinaSRImageProcessor


else:
   import sys

   lowercase__					  =		_LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__)

