[BugFix] Fix qwen matrix alignment (#7473)

WAI-clear · web-flow · commit 48bee12c4906 · 2024-01-02T11:35:01.000+08:00
diff --git a/paddlenlp/transformers/qwen/modeling.py b/paddlenlp/transformers/qwen/modeling.py
@@ -507,7 +507,7 @@ def _get_name_mappings(cls, config: QWenConfig) -> List[StateDictNameMapping]:
                 mapping[1] = "qwen." + mapping[1]
 
         if config.architectures is not None:
-            if "QWenForCausalLM" or "QWenLMHeadModel" in config.architectures:
+            if "QWenForCausalLM" in config.architectures or "QWenLMHeadModel" in config.architectures:
                 mappings.extend(
                     [
                         [

Original file line number	Diff line number	Diff line change
`@@ -507,7 +507,7 @@ def _get_name_mappings(cls, config: QWenConfig) -> List[StateDictNameMapping]:`
`507`	`507`	`mapping[1] = "qwen." + mapping[1]`
`508`	`508`
`509`	`509`	`if config.architectures is not None:`
`510`		`- if "QWenForCausalLM" or "QWenLMHeadModel" in config.architectures:`
	`510`	`+ if "QWenForCausalLM" in config.architectures or "QWenLMHeadModel" in config.architectures:`
`511`	`511`	`mappings.extend(`
`512`	`512`	`[`
`513`	`513`	`[`