Skip to content

Commit

Permalink
skip eetq tests loading shard_checkpoint
Browse files Browse the repository at this point in the history
  • Loading branch information
MekkCyber committed Nov 21, 2024
1 parent b2e08a8 commit 4a7158d
Showing 1 changed file with 11 additions and 0 deletions.
11 changes: 11 additions & 0 deletions src/transformers/utils/import_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -1006,6 +1006,17 @@ def is_auto_gptq_available():


def is_eetq_available():
if not _eetq_available:
return _eetq_available

try:
from eetq import EetqLinear # noqa: F401
except ImportError as exc:
if "shard_checkpoint" in str(exc):
# eetq is currently broken with newer transformers versions because it tries to import shard_checkpoint
# see https://github.com/NetEase-FuXi/EETQ/issues/34
# TODO: Remove once eetq releasees a fix and this release is used in CI
return False
return _eetq_available


Expand Down

0 comments on commit 4a7158d

Please sign in to comment.