Explorar el Código

Handle null rope scaling value (#2793)

Nigel Bosch hace 2 años
padre
commit
a2ca4e9de9
Se han modificado 1 ficheros con 2 adiciones y 1 borrados
  1. 2 1
      convert.py

+ 2 - 1
convert.py

@@ -170,7 +170,8 @@ class Params:
         f_norm_eps       = config["rms_norm_eps"]
         f_rope_freq_base = config["rope_theta"] if "rope_theta" in config else None
 
-        if "rope_scaling" in config and config["rope_scaling"].get("type") == "linear":
+        rope_scaling = config.get("rope_scaling")
+        if isinstance(rope_scaling, dict) and rope_scaling.get("type") == "linear":
             f_rope_scale = config["rope_scaling"].get("factor")
         else:
             f_rope_scale = None