'''simple docstring'''






from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available


_UpperCAmelCase    :     int         =       {
    '''configuration_megatron_bert''': ['''MEGATRON_BERT_PRETRAINED_CONFIG_ARCHIVE_MAP''', '''MegatronBertConfig'''],
}

try:
   if not is_torch_available():
      raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
   pass
else:
   _UpperCAmelCase    :     List[str]         =       [
       '''MEGATRON_BERT_PRETRAINED_MODEL_ARCHIVE_LIST''',
       '''MegatronBertForCausalLM''',
       '''MegatronBertForMaskedLM''',
       '''MegatronBertForMultipleChoice''',
       '''MegatronBertForNextSentencePrediction''',
       '''MegatronBertForPreTraining''',
       '''MegatronBertForQuestionAnswering''',
       '''MegatronBertForSequenceClassification''',
       '''MegatronBertForTokenClassification''',
       '''MegatronBertModel''',
       '''MegatronBertPreTrainedModel''',
   ]

if TYPE_CHECKING:
   from .configuration_megatron_bert import MEGATRON_BERT_PRETRAINED_CONFIG_ARCHIVE_MAP, MegatronBertConfig

   try:
      if not is_torch_available():
         raise OptionalDependencyNotAvailable()
   except OptionalDependencyNotAvailable:
      pass
   else:
      from .modeling_megatron_bert import (
          MEGATRON_BERT_PRETRAINED_MODEL_ARCHIVE_LIST,
          MegatronBertForCausalLM,
          MegatronBertForMaskedLM,
          MegatronBertForMultipleChoice,
          MegatronBertForNextSentencePrediction,
          MegatronBertForPreTraining,
          MegatronBertForQuestionAnswering,
          MegatronBertForSequenceClassification,
          MegatronBertForTokenClassification,
          MegatronBertModel,
          MegatronBertPreTrainedModel,
      )

else:
   import sys

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




'''simple docstring'''






import platform
from argparse import ArgumentParser

import huggingface_hub

from .. import __version__ as version
from ..utils import is_accelerate_available, is_torch_available, is_transformers_available, is_xformers_available
from . import BaseDiffusersCLICommand





def  _SCREAMING_SNAKE_CASE		(      __snake_case							:			Union[str, Any]				):
     return EnvironmentCommand()







class lowercase_       (		_UpperCamelCase			):







   """simple docstring"""

   @staticmethod
   def 	__UpperCAmelCase   (      UpperCamelCase__      :			ArgumentParser     )  ->			List[str]:
        _A    		=  parser.add_parser('env'     )
        download_parser.set_defaults(func=UpperCamelCase__     )
   def 	__UpperCAmelCase   (      self      :			Optional[int]     )  ->			Optional[Any]:
        _A    		=  huggingface_hub.__version__

        _A    		=  'not installed'
        _A    		=  'NA'
        if is_torch_available():
             import torch

             _A    		=  torch.__version__
             _A    		=  torch.cuda.is_available()

        _A    		=  'not installed'
        if is_transformers_available():
             import transformers

             _A    		=  transformers.__version__

        _A    		=  'not installed'
        if is_accelerate_available():
             import accelerate

             _A    		=  accelerate.__version__

        _A    		=  'not installed'
        if is_xformers_available():
             import xformers

             _A    		=  xformers.__version__

        _A    		=  {
            '`diffusers` version': version,
            'Platform': platform.platform(),
            'Python version': platform.python_version(),
            'PyTorch version (GPU?)': f'{pt_version} ({pt_cuda_available})',
            'Huggingface_hub version': hub_version,
            'Transformers version': transformers_version,
            'Accelerate version': accelerate_version,
            'xFormers version': xformers_version,
            'Using GPU in script?': '<fill in>',
            'Using distributed or parallel set-up in script?': '<fill in>',
        }

        print('\nCopy-and-paste the text below in your GitHub issue and FILL OUT the two last points.\n'     )
        print(self.format_dict(UpperCamelCase__     )     )

        return info

   @staticmethod
   def 	__UpperCAmelCase   (      UpperCamelCase__      :			List[Any]     )  ->			Dict:
        return "\n".join([f'- {prop}: {val}' for prop, val in d.items()]     ) + "\n"



