






"""simple docstring"""





import copy

from ...configuration_utils import PretrainedConfig
from ...utils import logging


UpperCamelCase					=     logging.get_logger(__name__)

UpperCamelCase					=     {
    """facebook/sam-vit-huge""": """https://huggingface.co/facebook/sam-vit-huge/resolve/main/config.json""",
    """facebook/sam-vit-large""": """https://huggingface.co/facebook/sam-vit-large/resolve/main/config.json""",
    """facebook/sam-vit-base""": """https://huggingface.co/facebook/sam-vit-base/resolve/main/config.json""",
}







class 						_a  (		lowercase_       ):





   '''simple docstring'''






   def __init__(       self					, UpperCAmelCase_=256					, UpperCAmelCase_=1_024					, UpperCAmelCase_=16					, UpperCAmelCase_=16					, UpperCAmelCase_=4					, UpperCAmelCase_="gelu"					, UpperCAmelCase_=1E-6					, **UpperCAmelCase_					, )			->   Any:




          '''simple docstring'''




          super().__init__(**UpperCAmelCase_)
          lowercase__:							Dict         =     hidden_size
          lowercase__:							List[Any]         =     image_size
          lowercase__:							int         =     patch_size
          lowercase__:							Optional[Any]         =     image_size // patch_size
          lowercase__:							str         =     mask_input_channels
          lowercase__:							Any         =     num_point_embeddings
          lowercase__:							Optional[Any]         =     hidden_act
          lowercase__:							List[str]         =     layer_norm_eps







class 						_a  (		lowercase_       ):





   '''simple docstring'''






   def __init__(       self					, UpperCAmelCase_=256					, UpperCAmelCase_="relu"					, UpperCAmelCase_=2_048					, UpperCAmelCase_=2					, UpperCAmelCase_=8					, UpperCAmelCase_=2					, UpperCAmelCase_=3					, UpperCAmelCase_=3					, UpperCAmelCase_=256					, UpperCAmelCase_=1E-6					, **UpperCAmelCase_					, )			->   Optional[Any]:




          '''simple docstring'''




          super().__init__(**UpperCAmelCase_)
          lowercase__:							Dict         =     hidden_size
          lowercase__:							Dict         =     hidden_act
          lowercase__:							List[str]         =     mlp_dim
          lowercase__:							List[str]         =     num_hidden_layers
          lowercase__:							List[str]         =     num_attention_heads
          lowercase__:							int         =     attention_downsample_rate
          lowercase__:							Optional[int]         =     num_multimask_outputs
          lowercase__:							str         =     iou_head_depth
          lowercase__:							List[Any]         =     iou_head_hidden_dim
          lowercase__:							List[str]         =     layer_norm_eps







class 						_a  (		lowercase_       ):





   '''simple docstring'''






   def __init__(       self					, UpperCAmelCase_=768					, UpperCAmelCase_=256					, UpperCAmelCase_=12					, UpperCAmelCase_=12					, UpperCAmelCase_=3					, UpperCAmelCase_=1_024					, UpperCAmelCase_=16					, UpperCAmelCase_="gelu"					, UpperCAmelCase_=1E-0_6					, UpperCAmelCase_=0.0					, UpperCAmelCase_=1E-1_0					, UpperCAmelCase_=True					, UpperCAmelCase_=4.0					, UpperCAmelCase_=True					, UpperCAmelCase_=True					, UpperCAmelCase_=14					, UpperCAmelCase_=[2, 5, 8, 11]					, UpperCAmelCase_=128					, UpperCAmelCase_=None					, **UpperCAmelCase_					, )			->   Any:




          '''simple docstring'''




          super().__init__(**UpperCAmelCase_)

          lowercase__:							Tuple         =     hidden_size
          lowercase__:							Union[str, Any]         =     output_channels
          lowercase__:							List[str]         =     num_hidden_layers
          lowercase__:							Any         =     num_attention_heads
          lowercase__:							Optional[int]         =     num_channels
          lowercase__:							List[Any]         =     image_size
          lowercase__:							Dict         =     patch_size
          lowercase__:							Tuple         =     hidden_act
          lowercase__:							Union[str, Any]         =     layer_norm_eps
          lowercase__:							Dict         =     attention_dropout
          lowercase__:							List[str]         =     initializer_range
          lowercase__:							List[str]         =     qkv_bias
          lowercase__:							Tuple         =     mlp_ratio
          lowercase__:							Optional[int]         =     use_abs_pos
          lowercase__:							Dict         =     use_rel_pos
          lowercase__:							List[str]         =     window_size
          lowercase__:							Optional[int]         =     global_attn_indexes
          lowercase__:							List[Any]         =     num_pos_feats
          lowercase__:							List[Any]         =     int(hidden_size * mlp_ratio) if mlp_dim is None else mlp_dim







class 						_a  (		lowercase_       ):





   '''simple docstring'''






   UpperCamelCase__     	=							"""sam"""
   UpperCamelCase__     	=							True





   def __init__(       self					, UpperCAmelCase_=None					, UpperCAmelCase_=None					, UpperCAmelCase_=None					, UpperCAmelCase_=0.02					, **UpperCAmelCase_					, )			->   Optional[int]:




          '''simple docstring'''




          super().__init__(**UpperCAmelCase_)
          lowercase__:							Tuple         =     vision_config if vision_config is not None else {}
          lowercase__:							Optional[int]         =     prompt_encoder_config if prompt_encoder_config is not None else {}
          lowercase__:							Dict         =     mask_decoder_config if mask_decoder_config is not None else {}

          if isinstance(UpperCAmelCase_					, UpperCAmelCase_):
                 lowercase__:							str         =     vision_config.to_dict()
          if isinstance(UpperCAmelCase_					, UpperCAmelCase_):
                 lowercase__:							List[str]         =     prompt_encoder_config.to_dict()
          if isinstance(UpperCAmelCase_					, UpperCAmelCase_):
                 lowercase__:							List[str]         =     mask_decoder_config.to_dict()

          lowercase__:							int         =     SamVisionConfig(**UpperCAmelCase_)
          lowercase__:							Optional[int]         =     SamPromptEncoderConfig(**UpperCAmelCase_)
          lowercase__:							Optional[Any]         =     SamMaskDecoderConfig(**UpperCAmelCase_)
          lowercase__:							Optional[int]         =     initializer_range





   def __lowercase							(       self)			->   Optional[Any]:




          '''simple docstring'''




          lowercase__:							List[Any]         =     copy.deepcopy(self.__dict__)
          lowercase__:							Any         =     self.vision_config.to_dict()
          lowercase__:							int         =     self.prompt_encoder_config.to_dict()
          lowercase__:							List[Any]         =     self.mask_decoder_config.to_dict()
          lowercase__:							Any         =     self.__class__.model_type
          return output












from typing import TYPE_CHECKING

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


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

try:
	if not is_torch_available():
		raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
	pass
else:
	__lowerCamelCase :    Any      						=      [
	    '''Constraint''',
	    '''ConstraintListState''',
	    '''DisjunctiveConstraint''',
	    '''PhrasalConstraint''',
	]
	__lowerCamelCase :    int      						=      [
	    '''BeamHypotheses''',
	    '''BeamScorer''',
	    '''BeamSearchScorer''',
	    '''ConstrainedBeamSearchScorer''',
	]
	__lowerCamelCase :    Union[str, 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''',
	]
	__lowerCamelCase :    Union[str, Any]      						=      [
	    '''MaxNewTokensCriteria''',
	    '''MaxLengthCriteria''',
	    '''MaxTimeCriteria''',
	    '''StoppingCriteria''',
	    '''StoppingCriteriaList''',
	    '''validate_stopping_criteria''',
	]
	__lowerCamelCase :    Union[str, 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:
	__lowerCamelCase :    Tuple      						=      [
	    '''TFForcedBOSTokenLogitsProcessor''',
	    '''TFForcedEOSTokenLogitsProcessor''',
	    '''TFLogitsProcessor''',
	    '''TFLogitsProcessorList''',
	    '''TFLogitsWarper''',
	    '''TFMinLengthLogitsProcessor''',
	    '''TFNoBadWordsLogitsProcessor''',
	    '''TFNoRepeatNGramLogitsProcessor''',
	    '''TFRepetitionPenaltyLogitsProcessor''',
	    '''TFTemperatureLogitsWarper''',
	    '''TFTopKLogitsWarper''',
	    '''TFTopPLogitsWarper''',
	    '''TFForceTokensLogitsProcessor''',
	    '''TFSuppressTokensAtBeginLogitsProcessor''',
	    '''TFSuppressTokensLogitsProcessor''',
	]
	__lowerCamelCase :    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:
	__lowerCamelCase :    Optional[int]      						=      [
	    '''FlaxForcedBOSTokenLogitsProcessor''',
	    '''FlaxForcedEOSTokenLogitsProcessor''',
	    '''FlaxLogitsProcessor''',
	    '''FlaxLogitsProcessorList''',
	    '''FlaxLogitsWarper''',
	    '''FlaxMinLengthLogitsProcessor''',
	    '''FlaxTemperatureLogitsWarper''',
	    '''FlaxTopKLogitsWarper''',
	    '''FlaxTopPLogitsWarper''',
	]
	__lowerCamelCase :    Union[str, Any]      						=      [
	    '''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 :    int      						=      _LazyModule(__name__, globals()['''__file__'''], _import_structure, module_spec=__spec__)



