
'''simple docstring'''







from dataclasses import dataclass
from enum import Enum
from typing import List, Optional, Union

import numpy as np
import PIL
from PIL import Image

from ...utils import BaseOutput, is_torch_available, is_transformers_available


@dataclass
class 			lowerCAmelCase (				__UpperCAmelCase			):
		a							:					Union[List[PIL.Image.Image], np.ndarray]
		a							:					Optional[List[bool]]


if is_transformers_available() and is_torch_available():
						from .pipeline_semantic_stable_diffusion import SemanticStableDiffusionPipeline

'''simple docstring'''







from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_tf_available, is_torch_available


_snake_case				:   Optional[Any]     =				{
    """configuration_rag""": ["""RagConfig"""],
    """retrieval_rag""": ["""RagRetriever"""],
    """tokenization_rag""": ["""RagTokenizer"""],
}

try:
						if not is_torch_available():
												raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
						pass
else:
						_snake_case				:   int     =				[
						    """RagModel""",
						    """RagPreTrainedModel""",
						    """RagSequenceForGeneration""",
						    """RagTokenForGeneration""",
						]

try:
						if not is_tf_available():
												raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
						pass
else:
						_snake_case				:   Dict     =				[
						    """TFRagModel""",
						    """TFRagPreTrainedModel""",
						    """TFRagSequenceForGeneration""",
						    """TFRagTokenForGeneration""",
						]


if TYPE_CHECKING:
						from .configuration_rag import RagConfig
						from .retrieval_rag import RagRetriever
						from .tokenization_rag import RagTokenizer

						try:
												if not is_torch_available():
																		raise OptionalDependencyNotAvailable()
						except OptionalDependencyNotAvailable:
												pass
						else:
												from .modeling_rag import RagModel, RagPreTrainedModel, RagSequenceForGeneration, RagTokenForGeneration

						try:
												if not is_tf_available():
																		raise OptionalDependencyNotAvailable()
						except OptionalDependencyNotAvailable:
												pass
						else:
												from .modeling_tf_rag import (
												    TFRagModel,
												    TFRagPreTrainedModel,
												    TFRagSequenceForGeneration,
												    TFRagTokenForGeneration,
												)

else:
						import sys

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