423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-05 13:12:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix log level

Browse Source 
Former-commit-id: 7fbe8add8f449358c9815c5ba8a2052a2d874dab
This commit is contained in:
hiyouga 2024-04-24 23:42:59 +08:00
parent ce490c65ae
commit 9a21785396
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)