Add vLLM patched image for K100: fix extra_special_tokens list->dict
This commit is contained in:
4
Dockerfile
Normal file
4
Dockerfile
Normal file
@@ -0,0 +1,4 @@
|
|||||||
|
FROM git.modelhub.org.cn:9443/enginex-hygon/vllm:0.9.2
|
||||||
|
|
||||||
|
COPY patch.py /tmp/patch.py
|
||||||
|
RUN python3 /tmp/patch.py
|
||||||
19
patch.py
Normal file
19
patch.py
Normal file
@@ -0,0 +1,19 @@
|
|||||||
|
path = '/usr/local/lib/python3.10/dist-packages/transformers/tokenization_utils_base.py'
|
||||||
|
with open(path, 'r') as f:
|
||||||
|
content = f.read()
|
||||||
|
|
||||||
|
old = (' self.SPECIAL_TOKENS_ATTRIBUTES = self.SPECIAL_TOKENS_ATTRIBUTES'
|
||||||
|
' + list(special_tokens.keys())')
|
||||||
|
new = (' # PATCH: some models have extra_special_tokens as list instead of dict\n'
|
||||||
|
' if isinstance(special_tokens, list):\n'
|
||||||
|
' special_tokens = {t: t for t in special_tokens}\n'
|
||||||
|
' self.SPECIAL_TOKENS_ATTRIBUTES = self.SPECIAL_TOKENS_ATTRIBUTES'
|
||||||
|
' + list(special_tokens.keys())')
|
||||||
|
|
||||||
|
if old in content:
|
||||||
|
content = content.replace(old, new)
|
||||||
|
with open(path, 'w') as f:
|
||||||
|
f.write(content)
|
||||||
|
print('Patch 1 applied successfully')
|
||||||
|
else:
|
||||||
|
print('WARNING: Patch 1 pattern not found')
|
||||||
Reference in New Issue
Block a user