Unverified Commit b691dcc4 authored by JieXin Liang's avatar JieXin Liang Committed by GitHub
Browse files

[misc] reduce weird rope_scaling_factor warning (#7176)

parent 0c9c6c75
...@@ -148,11 +148,14 @@ class InternLM2Config(PretrainedConfig): ...@@ -148,11 +148,14 @@ class InternLM2Config(PretrainedConfig):
if ( if (
rope_scaling_factor is None rope_scaling_factor is None
or not isinstance(rope_scaling_factor, float) or not isinstance(rope_scaling_factor, float)
or not isinstance(rope_scaling_factor, int)
or rope_scaling_factor < 1.0 or rope_scaling_factor < 1.0
): ):
raise ValueError( raise ValueError(
f"`rope_scaling`'s factor field must be a float >= 1, got {rope_scaling_factor}" f"`rope_scaling`'s factor field must be a float|int >= 1, got {rope_scaling_factor}"
) )
if isinstance(rope_scaling_factor, int):
rope_scaling_factor = float(rope_scaling_factor)
class InternVisionConfig(PretrainedConfig): class InternVisionConfig(PretrainedConfig):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment