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):
if (
rope_scaling_factor is None
or not isinstance(rope_scaling_factor, float)
or not isinstance(rope_scaling_factor, int)
or rope_scaling_factor < 1.0
):
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):
......
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