




"""simple docstring"""

from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available


a         =      {
    '''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:
       a         =      [
           '''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

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







"""simple docstring"""

import os

from huggingface_hub.constants import HUGGINGFACE_HUB_CACHE, hf_cache_home


a         =      HUGGINGFACE_HUB_CACHE


a         =      '''config.json'''
a         =      '''diffusion_pytorch_model.bin'''
a         =      '''diffusion_flax_model.msgpack'''
a         =      '''model.onnx'''
a         =      '''diffusion_pytorch_model.safetensors'''
a         =      '''weights.pb'''
a         =      '''https://huggingface.co'''
a         =      default_cache_path
a         =      '''diffusers_modules'''
a         =      os.getenv('''HF_MODULES_CACHE''', os.path.join(hf_cache_home, '''modules'''))
a         =      ['''fp16''', '''non-ema''']
a         =      '''.self_attn'''

