ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update _utils.py

Browse source
This commit is contained in:
Daniel Han 2024-11-03 23:55:39 -08:00
commit 9bae6e2149
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
unsloth/models/_utils.py
View file

@ -100,8 +100,9 @@ class HideLoggingMessage(logging.Filter):
pass
# The speedups for torchdynamo mostly come wih GPU Ampere or higher and which is not detected here.
import transformers.training_args.logger
transformers.training_args.logger.addFilter(HideLoggingMessage("The speedups"))
from transformers.training_args import logger as transformers_training_args_logger
transformers_training_args_logger.addFilter(HideLoggingMessage("The speedups"))
del transformers_training_args_logger
# =============================================