From 226973a9c585d78f860a4fd6355d880c6671e18f Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Tue, 9 Feb 2021 21:43:41 +0300 Subject: [PATCH] fix import (#10103) --- src/transformers/file_utils.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/transformers/file_utils.py b/src/transformers/file_utils.py index 073557e870..53562e28f6 100644 --- a/src/transformers/file_utils.py +++ b/src/transformers/file_utils.py @@ -144,7 +144,11 @@ try: _faiss_version = importlib_metadata.version("faiss") logger.debug(f"Successfully imported faiss version {_faiss_version}") except importlib_metadata.PackageNotFoundError: - _faiss_available = False + try: + _faiss_version = importlib_metadata.version("faiss-cpu") + logger.debug(f"Successfully imported faiss version {_faiss_version}") + except importlib_metadata.PackageNotFoundError: + _faiss_available = False _scatter_available = importlib.util.find_spec("torch_scatter") is not None