
'''simple docstring'''







import inspect
import os
import unittest
from pathlib import Path

import torch

import accelerate
from accelerate.test_utils import execute_subprocess_async
from accelerate.test_utils.testing import run_command







class UpperCAmelCase__	( unittest.TestCase	):





  """simple docstring"""

  __UpperCAmelCase				:				Optional[Any]   =					inspect.getfile(accelerate.test_utils	)
  __UpperCAmelCase				:				List[str]   =					os.path.sep.join(mod_file.split(os.path.sep	)[:-1] + ['''scripts''', '''test_cli.py''']	)

  __UpperCAmelCase				:				Dict   =					['''accelerate''', '''launch''']
  __UpperCAmelCase				:				Dict   =					Path.home() / '''.cache/huggingface/accelerate'''
  __UpperCAmelCase				:				Dict   =					'''default_config.yaml'''
  __UpperCAmelCase				:				Optional[Any]   =					config_folder / config_file
  __UpperCAmelCase				:				Dict   =					config_folder / '''_default_config.yaml'''

  __UpperCAmelCase				:				Any   =					Path('''tests/test_configs'''	)


  @classmethod
  def   __lowercase		(		cls      :     int       ):


     '''simple docstring'''






     if cls.config_path.is_file():
        cls.config_path.rename(cls.changed_path       )


  @classmethod
  def   __lowercase		(		cls      :     List[Any]       ):


     '''simple docstring'''






     if cls.changed_path.is_file():
        cls.changed_path.rename(cls.config_path       )


  def   __lowercase		(		self      :     Optional[int]       ):


     '''simple docstring'''






     _a				:	Dict			=   self.base_cmd
     if torch.cuda.is_available() and (torch.cuda.device_count() > 1):
        cmd += ["--multi_gpu"]
     execute_subprocess_async(cmd + [self.test_file_path]							,env=os.environ.copy()       )


  def   __lowercase		(		self      :     Optional[Any]       ):


     '''simple docstring'''






     for config in sorted(self.test_config_path.glob('**/*.yaml'       )       ):
        with self.subTest(config_file=_a       ):
           execute_subprocess_async(
               self.base_cmd + ['--config_file', str(_a       ), self.test_file_path]							,env=os.environ.copy()       )


  def   __lowercase		(		self      :     Optional[int]       ):


     '''simple docstring'''






     execute_subprocess_async(['accelerate', 'test']							,env=os.environ.copy()       )







class UpperCAmelCase__	( unittest.TestCase	):





  """simple docstring"""

  __UpperCAmelCase				:				Optional[Any]   =					'''test-tpu'''
  __UpperCAmelCase				:				Any   =					'''us-central1-a'''
  __UpperCAmelCase				:				List[Any]   =					'''ls'''
  __UpperCAmelCase				:				Any   =					['''accelerate''', '''tpu-config''']
  __UpperCAmelCase				:				Dict   =					'''cd /usr/share'''
  __UpperCAmelCase				:				Any   =					'''tests/test_samples/test_command_file.sh'''
  __UpperCAmelCase				:				List[Any]   =					'''Running gcloud compute tpus tpu-vm ssh'''


  def   __lowercase		(		self      :     Dict       ):


     '''simple docstring'''






     _a				:	Optional[Any]			=   run_command(
         self.cmd
         + ['--command', self.command, '--tpu_zone', self.tpu_zone, '--tpu_name', self.tpu_name, '--debug']							,return_stdout=_a							,)
     self.assertIn(
         F"""{self.gcloud} test-tpu --zone us-central1-a --command {self.base_output}; ls --worker all"""							,_a							,)


  def   __lowercase		(		self      :     List[str]       ):


     '''simple docstring'''






     _a				:	Any			=   run_command(
         self.cmd
         + [
             '--config_file',
             'tests/test_configs/0_12_0.yaml',
             '--command',
             self.command,
             '--tpu_zone',
             self.tpu_zone,
             '--tpu_name',
             self.tpu_name,
             '--debug',
         ]							,return_stdout=_a							,)
     self.assertIn(
         F"""{self.gcloud} test-tpu --zone us-central1-a --command {self.base_output}; ls --worker all"""							,_a							,)


  def   __lowercase		(		self      :     List[str]       ):


     '''simple docstring'''






     _a				:	Optional[int]			=   run_command(
         self.cmd + ['--config_file', 'tests/test_configs/latest.yaml', '--debug']							,return_stdout=_a       )
     self.assertIn(
         F"""{self.gcloud} test-tpu --zone us-central1-a --command {self.base_output}; echo \"hello world\"; echo \"this is a second command\" --worker all"""							,_a							,)


  def   __lowercase		(		self      :     int       ):


     '''simple docstring'''






     _a				:	Optional[Any]			=   run_command(
         self.cmd + ['--config_file', 'tests/test_configs/latest.yaml', '--command', self.command, '--debug']							,return_stdout=_a							,)
     self.assertIn(
         F"""{self.gcloud} test-tpu --zone us-central1-a --command {self.base_output}; ls --worker all"""							,_a							,)


  def   __lowercase		(		self      :     str       ):


     '''simple docstring'''






     _a				:	List[str]			=   run_command(
         self.cmd
         + [
             '--config_file',
             'tests/test_configs/latest.yaml',
             '--command',
             self.command,
             '--command',
             'echo "Hello World"',
             '--debug',
         ]							,return_stdout=_a							,)
     self.assertIn(
         F"""{self.gcloud} test-tpu --zone us-central1-a --command {self.base_output}; ls; echo \"Hello World\" --worker all"""							,_a							,)


  def   __lowercase		(		self      :     Union[str, Any]       ):


     '''simple docstring'''






     _a				:	Any			=   run_command(
         self.cmd
         + ['--config_file', 'tests/test_configs/latest.yaml', '--command_file', self.command_file, '--debug']							,return_stdout=_a							,)
     self.assertIn(
         F"""{self.gcloud} test-tpu --zone us-central1-a --command {self.base_output}; echo \"hello world\"; echo \"this is a second command\" --worker all"""							,_a							,)


  def   __lowercase		(		self      :     Union[str, Any]       ):


     '''simple docstring'''






     _a				:	Union[str, Any]			=   run_command(
         self.cmd
         + [
             '--config_file',
             'tests/test_configs/0_12_0.yaml',
             '--command_file',
             self.command_file,
             '--tpu_zone',
             self.tpu_zone,
             '--tpu_name',
             self.tpu_name,
             '--debug',
         ]							,return_stdout=_a							,)
     self.assertIn(
         F"""{self.gcloud} test-tpu --zone us-central1-a --command {self.base_output}; echo \"hello world\"; echo \"this is a second command\" --worker all"""							,_a							,)


  def   __lowercase		(		self      :     Any       ):


     '''simple docstring'''






     _a				:	Optional[int]			=   run_command(
         self.cmd + ['--config_file', 'tests/test_configs/latest.yaml', '--install_accelerate', '--debug']							,return_stdout=_a							,)
     self.assertIn(
         F"""{self.gcloud} test-tpu --zone us-central1-a --command {self.base_output}; pip install accelerate -U; echo \"hello world\"; echo \"this is a second command\" --worker all"""							,_a							,)


  def   __lowercase		(		self      :     List[str]       ):


     '''simple docstring'''






     _a				:	Optional[int]			=   run_command(
         self.cmd
         + [
             '--config_file',
             'tests/test_configs/latest.yaml',
             '--install_accelerate',
             '--accelerate_version',
             '12.0.0',
             '--debug',
         ]							,return_stdout=_a							,)
     self.assertIn(
         F"""{self.gcloud} test-tpu --zone us-central1-a --command {self.base_output}; pip install accelerate==12.0.0; echo \"hello world\"; echo \"this is a second command\" --worker all"""							,_a							,)





'''simple docstring'''







import os

from huggingface_hub.constants import HUGGINGFACE_HUB_CACHE, hf_cache_home


__lowerCAmelCase         =    HUGGINGFACE_HUB_CACHE


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



