





import torch




def 					A_(			):







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


if __name__ == "__main__":
				main()









import fire

from transformers import AutoConfig, AutoModelForSeqaSeqLM, AutoTokenizer




def    A_(			A				,      A				,      **A   ):
	UpperCAmelCase_								=     AutoConfig.from_pretrained(A				,      **A   )
	UpperCAmelCase_								=     AutoModelForSeqaSeqLM.from_config(A   )
	model.save_pretrained(A   )
	AutoTokenizer.from_pretrained(A   ).save_pretrained(A   )
	return model


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


