Skip to content

Commit 85e2f2d

Browse files
authored
Fresh export needs tokenizer patterns (#95)
1 parent 5624cef commit 85e2f2d

1 file changed

Lines changed: 9 additions & 1 deletion

File tree

python/src/coreai_models/vlm/export.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -260,7 +260,15 @@ async def export_text_bundle(
260260
logging.info(f"Downloading {spec.hf_model_id}...")
261261
model_dir = snapshot_download(
262262
spec.hf_model_id,
263-
allow_patterns=["*.safetensors", "*.safetensors.index.json", "config.json"],
263+
allow_patterns=[
264+
"*.safetensors",
265+
"*.safetensors.index.json",
266+
"config.json",
267+
"tokenizer*",
268+
"vocab.json",
269+
"merges.txt",
270+
"*.model",
271+
],
264272
)
265273
raw_cfg = AutoConfig.from_pretrained(model_dir)
266274
text_cfg = raw_cfg.text_config

0 commit comments

Comments
 (0)