Skip to content

Commit ceda77a

Browse files
committed
Extract out faiss index
1 parent d9b6141 commit ceda77a

File tree

3 files changed

+325
-268
lines changed

3 files changed

+325
-268
lines changed

bergson/__init__.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
1-
from .attributor import Attributor, FaissConfig
1+
from .attributor import Attributor
22
from .collection import collect_gradients
33
from .data import IndexConfig, load_gradients
4+
from .faiss_index import FaissConfig
45
from .gradcheck import FiniteDiff
56
from .gradients import (
67
GradientCollector,

0 commit comments

Comments
 (0)