Skip to content

Commit c95fae6

Browse files
patrickvonplatensgugger
authored andcommitted
fix import (#10103)
1 parent cc86472 commit c95fae6

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/transformers/file_utils.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,11 @@
144144
_faiss_version = importlib_metadata.version("faiss")
145145
logger.debug(f"Successfully imported faiss version {_faiss_version}")
146146
except importlib_metadata.PackageNotFoundError:
147-
_faiss_available = False
147+
try:
148+
_faiss_version = importlib_metadata.version("faiss-cpu")
149+
logger.debug(f"Successfully imported faiss version {_faiss_version}")
150+
except importlib_metadata.PackageNotFoundError:
151+
_faiss_available = False
148152

149153

150154
_scatter_available = importlib.util.find_spec("torch_scatter") is not None

0 commit comments

Comments
 (0)