





'''simple docstring'''

from typing import TYPE_CHECKING

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


UpperCAmelCase					:		Dict		=							{
    "configuration_groupvit": [
        "GROUPVIT_PRETRAINED_CONFIG_ARCHIVE_MAP",
        "GroupViTConfig",
        "GroupViTOnnxConfig",
        "GroupViTTextConfig",
        "GroupViTVisionConfig",
    ],
}

try:
   if not is_torch_available():
      raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
   pass
else:
   UpperCAmelCase					:		Optional[int]		=							[
       "GROUPVIT_PRETRAINED_MODEL_ARCHIVE_LIST",
       "GroupViTModel",
       "GroupViTPreTrainedModel",
       "GroupViTTextModel",
       "GroupViTVisionModel",
   ]

try:
   if not is_tf_available():
      raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
   pass
else:
   UpperCAmelCase					:		int		=							[
       "TF_GROUPVIT_PRETRAINED_MODEL_ARCHIVE_LIST",
       "TFGroupViTModel",
       "TFGroupViTPreTrainedModel",
       "TFGroupViTTextModel",
       "TFGroupViTVisionModel",
   ]

if TYPE_CHECKING:
   from .configuration_groupvit import (
       GROUPVIT_PRETRAINED_CONFIG_ARCHIVE_MAP,
       GroupViTConfig,
       GroupViTOnnxConfig,
       GroupViTTextConfig,
       GroupViTVisionConfig,
   )

   try:
      if not is_torch_available():
         raise OptionalDependencyNotAvailable()
   except OptionalDependencyNotAvailable:
      pass
   else:
      from .modeling_groupvit import (
          GROUPVIT_PRETRAINED_MODEL_ARCHIVE_LIST,
          GroupViTModel,
          GroupViTPreTrainedModel,
          GroupViTTextModel,
          GroupViTVisionModel,
      )

   try:
      if not is_tf_available():
         raise OptionalDependencyNotAvailable()
   except OptionalDependencyNotAvailable:
      pass
   else:
      from .modeling_tf_groupvit import (
          TF_GROUPVIT_PRETRAINED_MODEL_ARCHIVE_LIST,
          TFGroupViTModel,
          TFGroupViTPreTrainedModel,
          TFGroupViTTextModel,
          TFGroupViTVisionModel,
      )

else:
   import sys

   UpperCAmelCase					:		List[str]		=							_LazyModule(__name__, globals()['__file__'], _import_structure, module_spec=__spec__)









'''simple docstring'''

from ...utils import is_torch_available, is_transformers_available


if is_transformers_available() and is_torch_available():
   from .pipeline_vq_diffusion import LearnedClassifierFreeSamplingEmbeddings, VQDiffusionPipeline


