
"""simple docstring"""




from typing import TYPE_CHECKING

from ...utils import (
    OptionalDependencyNotAvailable,
    _LazyModule,
    is_sentencepiece_available,
    is_tokenizers_available,
    is_torch_available,
)


UpperCAmelCase__				:   int   							=			{
    'configuration_open_llama': ['OPEN_LLAMA_PRETRAINED_CONFIG_ARCHIVE_MAP', 'OpenLlamaConfig'],
}

try:
					if not is_sentencepiece_available():
										raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
					pass
else:
					UpperCAmelCase__				:   Union[str, Any]   							=			['LlamaTokenizer']

try:
					if not is_tokenizers_available():
										raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
					pass
else:
					UpperCAmelCase__				:   Dict   							=			['LlamaTokenizerFast']

try:
					if not is_torch_available():
										raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
					pass
else:
					UpperCAmelCase__				:   int   							=			[
					    'OpenLlamaForCausalLM',
					    'OpenLlamaModel',
					    'OpenLlamaPreTrainedModel',
					    'OpenLlamaForSequenceClassification',
					]


if TYPE_CHECKING:
					from .configuration_open_llama import OPEN_LLAMA_PRETRAINED_CONFIG_ARCHIVE_MAP, OpenLlamaConfig

					try:
										if not is_sentencepiece_available():
															raise OptionalDependencyNotAvailable()
					except OptionalDependencyNotAvailable:
										pass
					else:
										from transformers import LlamaTokenizer

					try:
										if not is_tokenizers_available():
															raise OptionalDependencyNotAvailable()
					except OptionalDependencyNotAvailable:
										pass
					else:
										from transformers import LlamaTokenizerFast

					try:
										if not is_torch_available():
															raise OptionalDependencyNotAvailable()
					except OptionalDependencyNotAvailable:
										pass
					else:
										from .modeling_open_llama import (
										    OpenLlamaForCausalLM,
										    OpenLlamaForSequenceClassification,
										    OpenLlamaModel,
										    OpenLlamaPreTrainedModel,
										)


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_torch_available


UpperCAmelCase__				:   Optional[Any]   							=			{'configuration_umt5': ['UMT5Config', 'UMT5OnnxConfig']}


try:
					if not is_torch_available():
										raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
					pass
else:
					UpperCAmelCase__				:   Tuple   							=			[
					    'UMT5EncoderModel',
					    'UMT5ForConditionalGeneration',
					    'UMT5ForQuestionAnswering',
					    'UMT5Model',
					    'UMT5PreTrainedModel',
					]

if TYPE_CHECKING:
					from .configuration_umta import UMTaConfig, UMTaOnnxConfig

					try:
										if not is_torch_available():
															raise OptionalDependencyNotAvailable()
					except OptionalDependencyNotAvailable:
										pass
					else:
										from .modeling_umta import (
										    UMTaEncoderModel,
										    UMTaForConditionalGeneration,
										    UMTaForQuestionAnswering,
										    UMTaModel,
										    UMTaPreTrainedModel,
										)
else:
					import sys

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




