Skip to content

Commit bef31a8

Browse files
committed
Improved ColBERT examples [skip ci]
1 parent 123f743 commit bef31a8

File tree

3 files changed

+9
-0
lines changed

3 files changed

+9
-0
lines changed

examples/colbert/approximate.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,10 @@
44
from colbert.modeling.checkpoint import Checkpoint
55
from pgvector.psycopg import register_vector
66
import psycopg
7+
import warnings
8+
9+
# ignore warnings from colbert
10+
warnings.filterwarnings('ignore')
711

812
conn = psycopg.connect(dbname='pgvector_example', autocommit=True)
913

examples/colbert/exact.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,10 @@
22
from colbert.modeling.checkpoint import Checkpoint
33
from pgvector.psycopg import register_vector
44
import psycopg
5+
import warnings
6+
7+
# ignore warnings from colbert
8+
warnings.filterwarnings('ignore')
59

610
conn = psycopg.connect(dbname='pgvector_example', autocommit=True)
711

examples/colbert/requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
11
colbert-ai
22
pgvector
33
psycopg[binary]
4+
transformers==4.49.0

0 commit comments

Comments
 (0)