423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-14 23:58:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix log level

Browse Source 
Former-commit-id: 8d21302f6201b3f33c10f61f3559bd95be3363c2
This commit is contained in:
hiyouga 2024-04-24 23:42:59 +08:00
parent 83404c4fa9
commit 21fac4c98c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/llmtuner/model/utils/rope.py
View File

@ -30,7 +30,7 @@ def configure_rope(config: "PretrainedConfig", model_args: "ModelArguments", is_
current_max_length = getattr(config, "max_position_embeddings", None) current_max_length = getattr(config, "max_position_embeddings", None)
if current_max_length and model_args.model_max_length > current_max_length: if current_max_length and model_args.model_max_length > current_max_length:
logger.warning( logger.info(
"Enlarge max model length from {} to {}.".format(current_max_length, model_args.model_max_length) "Enlarge max model length from {} to {}.".format(current_max_length, model_args.model_max_length)
) )
setattr(config, "max_position_embeddings", model_args.model_max_length) setattr(config, "max_position_embeddings", model_args.model_max_length)