Forráskód Böngészése

Handle null rope scaling value (#2793)

Nigel Bosch 2 éve
szülő
commit
a2ca4e9de9
1 módosított fájl, 2 hozzáadás és 1 törlés
  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