

"""simple docstring"""





import torch




def   _A  (  )     ->						int:
   '''simple docstring'''


   if torch.cuda.is_available():
      __lowercase								=						torch.cuda.device_count()
   else:
      __lowercase								=						0
   print(F"""Successfully ran on {num_gpus} GPUs""")


if __name__ == "__main__":
  main()







"""simple docstring"""





import fire

from transformers import AutoConfig, AutoModelForSeqaSeqLM, AutoTokenizer




def   _A  (  UpperCamelCase_					:	str,      UpperCamelCase_					:	str,      **UpperCamelCase_					:	Optional[int])     ->						Tuple:
   '''simple docstring'''


   __lowercase								=						AutoConfig.from_pretrained(UpperCamelCase_,      **UpperCamelCase_)
   __lowercase								=						AutoModelForSeqaSeqLM.from_config(UpperCamelCase_)
   model.save_pretrained(UpperCamelCase_)
   AutoTokenizer.from_pretrained(UpperCamelCase_).save_pretrained(UpperCamelCase_)
   return model


if __name__ == "__main__":
   fire.Fire(save_randomly_initialized_version)




