tokenizer.predict(x,batch_size=900) 大批量predict 时如何显示进度条?

如题,
小白想问一下,tokenizer.predict(x:list ,batch_size=900) 过程如何弄个进度条显示进度?

tqdm的话好像用于 for loop。
但是写成for loop 又降低了GPU处理效率。