Skip to content

Commit

Permalink
fix deepspeed available detection (huggingface#26252)
Browse files Browse the repository at this point in the history
  • Loading branch information
fxmarty authored and parambharat committed Sep 26, 2023
1 parent 70d203b commit 53a04da
Showing 1 changed file with 11 additions and 1 deletion.
12 changes: 11 additions & 1 deletion src/transformers/integrations/deepspeed.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@
Integration with Deepspeed
"""

import importlib.metadata as importlib_metadata
import importlib.util
import weakref
from functools import partialmethod
Expand All @@ -32,7 +33,16 @@


def is_deepspeed_available():
return importlib.util.find_spec("deepspeed") is not None
package_exists = importlib.util.find_spec("deepspeed") is not None

# Check we're not importing a "deepspeed" directory somewhere but the actual library by trying to grab the version
# AND checking it has an author field in the metadata that is HuggingFace.
if package_exists:
try:
_ = importlib_metadata.metadata("deepspeed")
return True
except importlib_metadata.PackageNotFoundError:
return False


if is_accelerate_available() and is_deepspeed_available():
Expand Down

0 comments on commit 53a04da

Please sign in to comment.