Add a check to import_utils.py to allow for use of faiss_gpu installation (#37997)

Adding check to import_utils.py for faiss_gpu

Author: Fiona-Waters

src/transformers/utils/import_utils.py

@@ -131,7 +131,7 @@ def _is_package_available(pkg_name: str, return_version: bool = False) -> Union[
 _yt_dlp_available = importlib.util.find_spec("yt_dlp") is not None
 _datasets_available = _is_package_available("datasets")
 _detectron2_available = _is_package_available("detectron2")
-# We need to check both `faiss` and `faiss-cpu`.
+# We need to check `faiss`, `faiss-cpu` and `faiss-gpu`.
 _faiss_available = importlib.util.find_spec("faiss") is not None
 try:
     _faiss_version = importlib.metadata.version("faiss")
@@ -141,7 +141,11 @@ def _is_package_available(pkg_name: str, return_version: bool = False) -> Union[
         _faiss_version = importlib.metadata.version("faiss-cpu")
         logger.debug(f"Successfully imported faiss version {_faiss_version}")
     except importlib.metadata.PackageNotFoundError:
-        _faiss_available = False
+        try:
+            _faiss_version = importlib.metadata.version("faiss-gpu")
+            logger.debug(f"Successfully imported faiss version {_faiss_version}")
+        except importlib.metadata.PackageNotFoundError:
+            _faiss_available = False
 _ftfy_available = _is_package_available("ftfy")
 _g2p_en_available = _is_package_available("g2p_en")
 _hadamard_available = _is_package_available("fast_hadamard_transform")