torch>=2.0.0 numpy>=1.21.0 tqdm>=4.64.0 argparse>=1.1 datasets transformers