



'''simple docstring'''





import inspect
import os
import sys
import unittest

import accelerate
from accelerate.test_utils import execute_subprocess_async, require_tpu







class     UpperCAmelCase_       (       unittest.TestCase	):
    """simple docstring"""







    def lowerCamelCase       (    self					:     Any						):
          snake_case__  :    Union[str, Any]    						=      inspect.getfile(accelerate.test_utils						)
          snake_case__  :    Optional[int]    						=      os.path.sep.join(mod_file.split(os.path.sep						)[:-1] + ["""scripts""", """test_script.py"""]						)
          snake_case__  :    Tuple    						=      os.path.sep.join(inspect.getfile(self.__class__						).split(os.path.sep						)[:-1]						)




    @require_tpu
    def lowerCamelCase       (    self					:     Tuple						):
          snake_case__  :    Optional[Any]    						=      f"\n            {self.test_dir}/xla_spawn.py\n            --num_cores 8\n            {self.test_file_path}\n        ".split()
          snake_case__  :    str    						=      [sys.executable] + distributed_args
          execute_subprocess_async(snake_case_ ,     env=os.environ.copy()						)





'''simple docstring'''





__a          =		frozenset(
    [
        "prompt",
        "height",
        "width",
        "guidance_scale",
        "negative_prompt",
        "prompt_embeds",
        "negative_prompt_embeds",
        "cross_attention_kwargs",
    ]
)

__a          =		frozenset(["prompt", "negative_prompt"])

__a          =		frozenset([])

__a          =		frozenset(["image"])

__a          =		frozenset(
    [
        "image",
        "height",
        "width",
        "guidance_scale",
    ]
)

__a          =		frozenset(["image"])

__a          =		frozenset(
    [
        "prompt",
        "image",
        "height",
        "width",
        "guidance_scale",
        "negative_prompt",
        "prompt_embeds",
        "negative_prompt_embeds",
    ]
)

__a          =		frozenset(["prompt", "image", "negative_prompt"])

__a          =		frozenset(
    [
        # Text guided image variation with an image mask
        "prompt",
        "image",
        "mask_image",
        "height",
        "width",
        "guidance_scale",
        "negative_prompt",
        "prompt_embeds",
        "negative_prompt_embeds",
    ]
)

__a          =		frozenset(["prompt", "image", "mask_image", "negative_prompt"])

__a          =		frozenset(
    [
        # image variation with an image mask
        "image",
        "mask_image",
        "height",
        "width",
        "guidance_scale",
    ]
)

__a          =		frozenset(["image", "mask_image"])

__a          =		frozenset(
    [
        "example_image",
        "image",
        "mask_image",
        "height",
        "width",
        "guidance_scale",
    ]
)

__a          =		frozenset(["example_image", "image", "mask_image"])

__a          =		frozenset(["class_labels"])

__a          =		frozenset(["class_labels"])

__a          =		frozenset(["batch_size"])

__a          =		frozenset([])

__a          =		frozenset(["batch_size"])

__a          =		frozenset([])

__a          =		frozenset(
    [
        "prompt",
        "audio_length_in_s",
        "guidance_scale",
        "negative_prompt",
        "prompt_embeds",
        "negative_prompt_embeds",
        "cross_attention_kwargs",
    ]
)

__a          =		frozenset(["prompt", "negative_prompt"])
__a          =		frozenset(["input_tokens"])

__a          =		frozenset(["input_tokens"])
