Skip to content
Open
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion libs/infinity_emb/infinity_emb/transformer/acceleration.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
from infinity_emb.primitives import Device

if CHECK_OPTIMUM.is_available:
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

logic: CHECK_OPTIMUM.is_available may not correctly detect the new better_transformer package since it checks for 'optimum' module availability.

from optimum.bettertransformer import ( # type: ignore[import-untyped]
from better_transformer import ( # type: ignore[import-untyped]
BetterTransformer,
BetterTransformerManager,
)
Expand Down
Loading
Loading