Skip to content

Commit 252b390

Browse files
authored
fix qwen-vl lmdeploy (#1572)
1 parent 7e751d5 commit 252b390

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

swift/llm/utils/lmdeploy_utils.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
import time
66
from contextlib import contextmanager
77
from copy import deepcopy
8+
from functools import wraps
89
from queue import Queue
910
from threading import Thread
1011
from typing import Any, Dict, Iterator, List, Optional, Tuple, Union
@@ -16,7 +17,7 @@
1617
from lmdeploy.serve.async_engine import AsyncEngine
1718
from lmdeploy.serve.vl_async_engine import VLAsyncEngine
1819
from tqdm import tqdm
19-
from transformers import AutoConfig, GenerationConfig
20+
from transformers import AutoConfig, AutoTokenizer, GenerationConfig
2021

2122
from swift.utils import get_logger
2223
from .argument import InferArguments
@@ -69,7 +70,16 @@ def get_lmdeploy_engine(
6970
pipeline_kwargs['vision_config'] = vision_config
7071
logger.info(f'vision_config: {vision_config}')
7172

73+
_old_from_pretrained = AutoTokenizer.from_pretrained
74+
75+
@wraps(_old_from_pretrained)
76+
def _from_pretrained(self, *args, **kwargs):
77+
return tokenizer
78+
79+
AutoTokenizer.from_pretrained = _from_pretrained
7280
lmdeploy_engine = pipeline(model_dir, backend_config=backend_config, **pipeline_kwargs)
81+
AutoTokenizer.from_pretrained = _old_from_pretrained # recover
82+
7383
lmdeploy_engine.model_dir = model_dir
7484
lmdeploy_engine.model_type = model_type
7585
lmdeploy_engine.is_multimodal = is_multimodal

0 commit comments

Comments
 (0)