





from __future__ import annotations




def  lowerCAmelCase_ (							UpperCamelCase_					,		UpperCamelCase_  )     ->		list[list[int]]:
       UpperCamelCase_          =  []
       create_all_state(1					,		UpperCamelCase_					,		UpperCamelCase_					,		[]					,		UpperCamelCase_  )
       return result




def  lowerCAmelCase_ (							UpperCamelCase_					,		UpperCamelCase_					,		UpperCamelCase_					,		UpperCamelCase_					,		UpperCamelCase_					,		)     ->		None:
       if level == 0:
              total_list.append(current_list[:]  )
              return

       for i in range(UpperCamelCase_					,		total_number - level + 2  ):
              current_list.append(UpperCamelCase_  )
              create_all_state(i + 1					,		UpperCamelCase_					,		level - 1					,		UpperCamelCase_					,		UpperCamelCase_  )
              current_list.pop()




def  lowerCAmelCase_ (							UpperCamelCase_  )     ->		None:
       for i in total_list:
              print(*UpperCamelCase_  )


if __name__ == "__main__":
     _UpperCAmelCase        =				4
     _UpperCAmelCase        =				2
     _UpperCAmelCase        =				generate_all_combinations(n, k)
     print_all_state(total_list)








import jax.numpy as jnp

from ...utils import logging
from ..ta.modeling_flax_ta import FlaxTaEncoderModel, FlaxTaForConditionalGeneration, FlaxTaModel
from .configuration_mta import MTaConfig


_UpperCAmelCase        =				logging.get_logger(__name__)

_UpperCAmelCase        =				'T5Config'




def  lowerCAmelCase_ (							UpperCamelCase_					,		UpperCamelCase_					,		UpperCamelCase_  )     ->		jnp.ndarray:
       UpperCamelCase_          =  jnp.zeros_like(UpperCamelCase_  )
       UpperCamelCase_          =  shifted_input_ids.at[:, 1:].set(input_ids[:, :-1]  )
       UpperCamelCase_          =  shifted_input_ids.at[:, 0].set(UpperCamelCase_  )

       UpperCamelCase_          =  jnp.where(shifted_input_ids == -100					,		UpperCamelCase_					,		UpperCamelCase_  )
       return shifted_input_ids



class 							_UpperCamelCase					(	lowerCAmelCase_     ):
       _UpperCamelCase				:					Dict         =   '''mt5'''
       _UpperCamelCase				:					Union[str, Any]         =   MTaConfig



class 							_UpperCamelCase					(	lowerCAmelCase_     ):
       _UpperCamelCase				:					Dict         =   '''mt5'''
       _UpperCamelCase				:					Optional[int]         =   MTaConfig



class 							_UpperCamelCase					(	lowerCAmelCase_     ):
       _UpperCamelCase				:					Union[str, Any]         =   '''mt5'''
       _UpperCamelCase				:					List[Any]         =   MTaConfig

