[fix][RL] Fix DeepSeekV3ForCausalLM.post_load_weights for multiple update weight (#6265)

This commit is contained in:
Zilin Zhu
2025-05-29 16:28:10 -07:00
committed by GitHub
parent 73def253b5
commit 51cdd81f97
2 changed files with 47 additions and 14 deletions
+8
View File
@@ -2217,3 +2217,11 @@ def read_system_prompt_from_file(model_name: str) -> str:
except Exception:
# If anything fails, return empty string
return ""
def bind_or_assign(target, source):
if target is not None:
target.copy_(source)
return target
else:
return source