Skip to content

Commit

Permalink
isort
Browse files Browse the repository at this point in the history
  • Loading branch information
bclavie committed Feb 23, 2024
1 parent 00e555b commit 9cce361
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions ragatouille/models/colbert.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@
from colbert import Indexer, IndexUpdater, Searcher, Trainer
from colbert.infra import ColBERTConfig, Run, RunConfig
from colbert.modeling.checkpoint import Checkpoint

from ragatouille.models.base import LateInteractionModel

# TODO: Move all bsize related calcs to `_set_bsize()`
Expand Down

0 comments on commit 9cce361

Please sign in to comment.