| #!/usr/bin/env python3 | |
| import argparse | |
| import transformers | |
| if __name__ == "__main__": | |
| parser = argparse.ArgumentParser() | |
| parser.add_argument("input_path", type=str, help="Input directory") | |
| parser.add_argument("output_path", type=str, help="Output directory") | |
| args = parser.parse_args() | |
| robeczech = transformers.TFAutoModelWithLMHead.from_pretrained(args.input_path, from_pt=True) | |
| robeczech.save_pretrained(args.output_path) | |