


'''simple docstring'''

from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available


_UpperCAmelCase				: Optional[Any]							    =				{
    """configuration_jukebox""": [
        """JUKEBOX_PRETRAINED_CONFIG_ARCHIVE_MAP""",
        """JukeboxConfig""",
        """JukeboxPriorConfig""",
        """JukeboxVQVAEConfig""",
    ],
    """tokenization_jukebox""": ["""JukeboxTokenizer"""],
}

try:
	if not is_torch_available():
		raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
	pass
else:
	_UpperCAmelCase				: Optional[Any]							    =				[
	    """JUKEBOX_PRETRAINED_MODEL_ARCHIVE_LIST""",
	    """JukeboxModel""",
	    """JukeboxPreTrainedModel""",
	    """JukeboxVQVAE""",
	    """JukeboxPrior""",
	]

if TYPE_CHECKING:
	from .configuration_jukebox import (
	    JUKEBOX_PRETRAINED_CONFIG_ARCHIVE_MAP,
	    JukeboxConfig,
	    JukeboxPriorConfig,
	    JukeboxVQVAEConfig,
	)
	from .tokenization_jukebox import JukeboxTokenizer

	try:
		if not is_torch_available():
			raise OptionalDependencyNotAvailable()
	except OptionalDependencyNotAvailable:
		pass
	else:
		from .modeling_jukebox import (
		    JUKEBOX_PRETRAINED_MODEL_ARCHIVE_LIST,
		    JukeboxModel,
		    JukeboxPreTrainedModel,
		    JukeboxPrior,
		    JukeboxVQVAE,
		)

else:
	import sys

	_UpperCAmelCase				: Union[str, Any]							    =				_LazyModule(__name__, globals()["""__file__"""], _import_structure, module_spec=__spec__)





'''simple docstring'''

from .glue import GlueDataset, GlueDataTrainingArguments
from .language_modeling import (
    LineByLineTextDataset,
    LineByLineWithRefDataset,
    LineByLineWithSOPTextDataset,
    TextDataset,
    TextDatasetForNextSentencePrediction,
)
from .squad import SquadDataset, SquadDataTrainingArguments

