AnswerDotAI / rerankers

A lightweight, low-dependency, unified API to use all common reranking and cross-encoder models.
Apache License 2.0
1.04k stars 57 forks source link

Suppress print message #28

Open gabrielchua opened 1 month ago

gabrielchua commented 1 month ago

How do I suppress the following?

Thank you!

Loading ColBERTRanker model answerdotai/answerai-colbert-small-v1
No device set
Using device mps
No dtype set
Using dtype torch.float16
Loading model answerdotai/answerai-colbert-small-v1, this might take a while...
Linear Dim set to: {linear_dim} for downcasting