Update _utils.py
This commit is contained in:
parent
d17519d819
commit
935754dba2
1 changed files with 8 additions and 0 deletions
|
|
@ -190,6 +190,14 @@ try:
|
|||
except:
|
||||
pass
|
||||
|
||||
# The following generation flags are not valid and may be ignored:
|
||||
try:
|
||||
from transformers.generation.configuration_utils import logger as configuration_logger
|
||||
configuration_logger.addFilter(HideLoggingMessage("following generation flags"))
|
||||
del configuration_logger
|
||||
except:
|
||||
pass
|
||||
|
||||
# Gemma3 It is strongly recommended to train Gemma3 models with the `eager`
|
||||
try:
|
||||
from transformers.models.gemma3.modeling_gemma3 import logger as gemma3_logger
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue