#!/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)