Skip to content

Commit

Permalink
chore: Pass self.config.avoid_fork_if_possible = True
Browse files Browse the repository at this point in the history
  • Loading branch information
bclavie committed Jan 14, 2024
1 parent 47869d8 commit 5351006
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions ragatouille/models/colbert.py
Original file line number Diff line number Diff line change
Expand Up @@ -192,6 +192,9 @@ def index(
self.config = ColBERTConfig.from_existing(
self.config, ColBERTConfig(nbits=nbits)
)

# Instruct colbert-ai to disable forking if nranks == 1
self.config.avoid_fork_if_possible = True
self.indexer = Indexer(
checkpoint=self.checkpoint,
config=self.config,
Expand Down

0 comments on commit 5351006

Please sign in to comment.