Forráskód Böngészése

Fix unboundlocalerror (#4641)

Lin Manhui 4 hete
szülő
commit
fb83c7acf2

+ 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,