FlagOpen
How to disable the progress bar
from FlagEmbedding import FlagLLMReranker reranker = FlagLLMReranker('/bge-reranker-v2-gemma/bge-reranker-v2-gemma', use_fp16=True) def reranker_embedding(sentences_1,sentences_2): score = reranker.compute_score([sentences_1,sentences_2],normalize=True) return score[0] import os os.environ["TOKENIZERS_PARALLELISM"] = "false" os.environ["DISABLE_TQDM"] = "true"
reranker_embedding('JOYTIME (HK) INTERNATIONAL COMPANY','Voit')
一直显示下面的进度条，如何禁用呢？ 100%|██████████| 1/1 [00:00<00:00, 47.67it/s]
这里需要修改下compute_score的源码，把trange换成range，把tqdm去掉
对于 LayerWiseFlagLLMReranker
需要修改的地方位于FlagEmbedding\inference\reranker\decoder_only\layerwise.py: 326，其他类型的模型类似
另外，是否可以添加配置项来选择是否使用tqdm?
@545999961
