瀏覽代碼

Fix unboundlocalerror (#4641)

Lin Manhui 4 周之前
父節點
當前提交
fb83c7acf2
共有 1 個文件被更改,包括 4 次插入0 次删除
  1. 4 0
      paddlex/inference/models/common/vlm/transformers/model_utils.py

+ 4 - 0
paddlex/inference/models/common/vlm/transformers/model_utils.py

@@ -1607,6 +1607,8 @@ class PretrainedModel(
                 except NotImplementedError:
                     if convert_from_hf:
                         raise ValueError("`convert_from_hf=True` is not supported")
+                    else:
+                        transpose_weight_keys = None
                 state_dict = load_state_dict(
                     shard_file,
                     tp_actions if pre_tensor_parallel_split else None,
@@ -1934,6 +1936,8 @@ class PretrainedModel(
                 except NotImplementedError:
                     if convert_from_hf:
                         raise ValueError("`convert_from_hf=True` is not supported")
+                    else:
+                        transpose_weight_keys = None
                 state_dict = load_state_dict(
                     resolved_archive_file,
                     convert_from_hf=convert_from_hf,