Skip to content

Commit

Permalink
add warnign deprecation bettertransformer
Browse files Browse the repository at this point in the history
  • Loading branch information
echarlaix committed Sep 27, 2024
1 parent 3572a0b commit d6e97cf
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions optimum/bettertransformer/transformation.py
Original file line number Diff line number Diff line change
Expand Up @@ -206,6 +206,10 @@ def transform(
The converted model if the conversion has been successful.
"""

logger.warning(
"The class `optimum.bettertransformers.transformation.BetterTransformer` is deprecated and will be removed in a future release."
)

hf_config = model.config
if hf_config.model_type in ["falcon", "gpt_bigcode", "llama", "whisper"]:
raise ValueError(
Expand Down

0 comments on commit d6e97cf

Please sign in to comment.