transformers matplotlib tqdm torch