[assets] update wechat (#9129)

This commit is contained in:
Yaowei Zheng
2025-09-14 03:05:08 +08:00
committed by GitHub
parent 52488ac974
commit 260b5625c3
3 changed files with 10 additions and 9 deletions

View File

@@ -219,7 +219,7 @@ def patch_valuehead_model(model: "AutoModelForCausalLMWithValueHead") -> None:
if base_model and hasattr(base_model, "get_rope_index"):
return base_model.get_rope_index
elif (base_model and hasattr(base_model, "model") and hasattr(base_model.model, "get_rope_index")):
elif base_model and hasattr(base_model, "model") and hasattr(base_model.model, "get_rope_index"):
return base_model.model.get_rope_index
else:
return None