

from typing import TYPE_CHECKING

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


a_     =			{
    'configuration_gpt_neo': ['GPT_NEO_PRETRAINED_CONFIG_ARCHIVE_MAP', 'GPTNeoConfig', 'GPTNeoOnnxConfig'],
}

try:
			if not is_torch_available():
						raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
			pass
else:
			a_     =			[
			    'GPT_NEO_PRETRAINED_MODEL_ARCHIVE_LIST',
			    'GPTNeoForCausalLM',
			    'GPTNeoForQuestionAnswering',
			    'GPTNeoForSequenceClassification',
			    'GPTNeoForTokenClassification',
			    'GPTNeoModel',
			    'GPTNeoPreTrainedModel',
			    'load_tf_weights_in_gpt_neo',
			]

try:
			if not is_flax_available():
						raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
			pass
else:
			a_     =			[
			    'FlaxGPTNeoForCausalLM',
			    'FlaxGPTNeoModel',
			    'FlaxGPTNeoPreTrainedModel',
			]


if TYPE_CHECKING:
			from .configuration_gpt_neo import GPT_NEO_PRETRAINED_CONFIG_ARCHIVE_MAP, GPTNeoConfig, GPTNeoOnnxConfig

			try:
						if not is_torch_available():
									raise OptionalDependencyNotAvailable()
			except OptionalDependencyNotAvailable:
						pass
			else:
						from .modeling_gpt_neo import (
						    GPT_NEO_PRETRAINED_MODEL_ARCHIVE_LIST,
						    GPTNeoForCausalLM,
						    GPTNeoForQuestionAnswering,
						    GPTNeoForSequenceClassification,
						    GPTNeoForTokenClassification,
						    GPTNeoModel,
						    GPTNeoPreTrainedModel,
						    load_tf_weights_in_gpt_neo,
						)

			try:
						if not is_flax_available():
									raise OptionalDependencyNotAvailable()
			except OptionalDependencyNotAvailable:
						pass
			else:
						from .modeling_flax_gpt_neo import FlaxGPTNeoForCausalLM, FlaxGPTNeoModel, FlaxGPTNeoPreTrainedModel


else:
			import sys

			a_     =			_LazyModule(__name__, globals()['__file__'], _import_structure, module_spec=__spec__)



from typing import TYPE_CHECKING

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


snake_case							:       str     			=      {
    '''configuration_rag''': ['''RagConfig'''],
    '''retrieval_rag''': ['''RagRetriever'''],
    '''tokenization_rag''': ['''RagTokenizer'''],
}

try:
						if not is_torch_available():
												raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
						pass
else:
						snake_case							:       Any     			=      [
						    '''RagModel''',
						    '''RagPreTrainedModel''',
						    '''RagSequenceForGeneration''',
						    '''RagTokenForGeneration''',
						]

try:
						if not is_tf_available():
												raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
						pass
else:
						snake_case							:       Optional[int]     			=      [
						    '''TFRagModel''',
						    '''TFRagPreTrainedModel''',
						    '''TFRagSequenceForGeneration''',
						    '''TFRagTokenForGeneration''',
						]


if TYPE_CHECKING:
						from .configuration_rag import RagConfig
						from .retrieval_rag import RagRetriever
						from .tokenization_rag import RagTokenizer

						try:
												if not is_torch_available():
																		raise OptionalDependencyNotAvailable()
						except OptionalDependencyNotAvailable:
												pass
						else:
												from .modeling_rag import RagModel, RagPreTrainedModel, RagSequenceForGeneration, RagTokenForGeneration

						try:
												if not is_tf_available():
																		raise OptionalDependencyNotAvailable()
						except OptionalDependencyNotAvailable:
												pass
						else:
												from .modeling_tf_rag import (
												    TFRagModel,
												    TFRagPreTrainedModel,
												    TFRagSequenceForGeneration,
												    TFRagTokenForGeneration,
												)

else:
						import sys

						snake_case							:       int     			=      _LazyModule(__name__, globals()['''__file__'''], _import_structure, module_spec=__spec__)




