support qwen3moe name_mapping (#2820)

This commit is contained in:
gaoziyuan
2025-07-12 12:05:54 +08:00
committed by GitHub
parent f6ad26fc08
commit 749b2e9c89
2 changed files with 98 additions and 1 deletions

View File

@@ -56,6 +56,7 @@ class ModelForCasualLM(nn.Layer, ABC):
ori_vocab_size, use_topp_sampling, etc.
"""
super(ModelForCasualLM, self).__init__()
self.fd_config = configs
@abstractmethod
def set_state_dict(self, state_dict: Dict[str, Union[np.ndarray,