





"""simple docstring"""
from typing import TYPE_CHECKING

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


__UpperCamelCase			:		Union[str, Any]		    =   {
    '''configuration_bark''': [
        '''BARK_PRETRAINED_CONFIG_ARCHIVE_MAP''',
        '''BarkCoarseConfig''',
        '''BarkConfig''',
        '''BarkFineConfig''',
        '''BarkSemanticConfig''',
    ],
    '''processing_bark''': ['''BarkProcessor'''],
}

try:
       if not is_torch_available():
              raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
       pass
else:
       __UpperCamelCase			:		str		    =   [
           '''BARK_PRETRAINED_MODEL_ARCHIVE_LIST''',
           '''BarkFineModel''',
           '''BarkSemanticModel''',
           '''BarkCoarseModel''',
           '''BarkModel''',
           '''BarkPreTrainedModel''',
           '''BarkCausalModel''',
       ]

if TYPE_CHECKING:
       from .configuration_bark import (
           BARK_PRETRAINED_CONFIG_ARCHIVE_MAP,
           BarkCoarseConfig,
           BarkConfig,
           BarkFineConfig,
           BarkSemanticConfig,
       )
       from .processing_bark import BarkProcessor

       try:
              if not is_torch_available():
                     raise OptionalDependencyNotAvailable()
       except OptionalDependencyNotAvailable:
              pass
       else:
              from .modeling_bark import (
                  BARK_PRETRAINED_MODEL_ARCHIVE_LIST,
                  BarkCausalModel,
                  BarkCoarseModel,
                  BarkFineModel,
                  BarkModel,
                  BarkPreTrainedModel,
                  BarkSemanticModel,
              )

else:
       import sys

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













"""simple docstring"""







import importlib.metadata
import subprocess
import sys





def 						lowercase__(A = False						)				->List[str]:




						"""simple docstring"""



						lowercase__				:      Any=       False
						if "IPython" in sys.modules:
												lowercase__				:      Tuple=       "google.colab" in str(sys.modules["IPython"].get_ipython()						)

						if in_colab:
												if upgrade:
																		lowercase__				:      Optional[int]=       ["pip", "install", "-U", "torch"]
																		subprocess.run(A					,		check=A						)
												# get the current version of torch
												lowercase__				:      Tuple=       importlib.metadata.version("torch"						)
												lowercase__				:      Dict=       torch_version[: torch_version.rindex("."						)]
												lowercase__				:      Dict=       f'''https://storage.googleapis.com/tpu-pytorch/wheels/colab/torch_xla-{torch_version_trunc}-cp37-cp37m-linux_x86_64.whl'''
												lowercase__				:      Union[str, Any]=       ["pip", "install", xla_wheel]
												subprocess.run(A					,		check=A						)
						else:
												raise RuntimeError("`install_xla` utility works only on google colab."						)






