Unverified Commit 233b26ab authored by zhang-prog's avatar zhang-prog Committed by GitHub
Browse files

[PaddleOCR-VL] Add BC for transformers 5.0 config (#33976)


Signed-off-by: default avatarzhangyue66 <zhangyue66@baidu.com>
parent 791a94be
...@@ -994,6 +994,13 @@ class PaddleOCRVLForConditionalGeneration(nn.Module, SupportsMultiModal, Support ...@@ -994,6 +994,13 @@ class PaddleOCRVLForConditionalGeneration(nn.Module, SupportsMultiModal, Support
def __init__(self, *, vllm_config: VllmConfig, prefix: str = ""): def __init__(self, *, vllm_config: VllmConfig, prefix: str = ""):
super().__init__() super().__init__()
config = vllm_config.model_config.hf_config config = vllm_config.model_config.hf_config
if hasattr(config, "text_config"):
text_config = config.text_config.to_dict()
unsafe_keys = ["model_type", "architectures", "tie_word_embeddings"]
for key in unsafe_keys:
text_config.pop(key, None)
config.update(text_config)
quant_config = vllm_config.quant_config quant_config = vllm_config.quant_config
self.config = config self.config = config
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment