File size: 450 Bytes
a75ff23 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
#!/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)
|