





from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available


UpperCamelCase__											= {
    "configuration_nllb_moe": [
        "NLLB_MOE_PRETRAINED_CONFIG_ARCHIVE_MAP",
        "NllbMoeConfig",
    ]
}

try:
					if not is_torch_available():
										raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
					pass
else:
					UpperCamelCase__											= [
					    "NLLB_MOE_PRETRAINED_MODEL_ARCHIVE_LIST",
					    "NllbMoeForConditionalGeneration",
					    "NllbMoeModel",
					    "NllbMoePreTrainedModel",
					    "NllbMoeTop2Router",
					    "NllbMoeSparseMLP",
					]


if TYPE_CHECKING:
					from .configuration_nllb_moe import (
					    NLLB_MOE_PRETRAINED_CONFIG_ARCHIVE_MAP,
					    NllbMoeConfig,
					)

					try:
										if not is_torch_available():
															raise OptionalDependencyNotAvailable()
					except OptionalDependencyNotAvailable:
										pass
					else:
										from .modeling_nllb_moe import (
										    NLLB_MOE_PRETRAINED_MODEL_ARCHIVE_LIST,
										    NllbMoeForConditionalGeneration,
										    NllbMoeModel,
										    NllbMoePreTrainedModel,
										    NllbMoeSparseMLP,
										    NllbMoeTopaRouter,
										)


else:
					import sys

					UpperCamelCase__											= _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__)











from argparse import ArgumentParser

from .env import EnvironmentCommand





def     _UpperCamelCase     ():






							"""simple docstring"""





							UpperCamelCase__				     =			ArgumentParser("""Diffusers CLI tool"""   ,			usage="""diffusers-cli <command> [<args>]"""       )
							UpperCamelCase__				     =			parser.add_subparsers(help="""diffusers-cli command helpers"""       )

							# Register commands
							EnvironmentCommand.register_subcommand(a__       )

							# Let's go
							UpperCamelCase__				     =			parser.parse_args()

							if not hasattr(a__   ,			"""func"""       ):
														parser.print_help()
														exit(1       )

							# Run
							UpperCamelCase__				     =			args.func(a__       )
							service.run()


if __name__ == "__main__":
					main()




