FlagEmbedding icon indicating copy to clipboard operation
FlagEmbedding copied to clipboard

allow HF model download, allow disabling tqdm bars

Open kylrth opened this issue 1 year ago • 0 comments

The first commit fixes a problem where the layerwise reranker classes could not download models from HF because local_files_only=True was set.

closes #721

The second commit adds as progress keyword argument to all reranker compute_score methods, which allows the caller to activate/disable the tqdm progress bar. Some of these methods previously disabled the progress bar if the number of samples was less than the arbitrary value of 128. Now all methods will deactivate it by default if there is only one batch, but the user can choose to set it on or off regardless by passing progress=True or progress=False.

kylrth avatar Aug 15 '24 00:08 kylrth