| 12345678910111213141516171819202122232425 |
- #!/usr/bin/env python3
- """Skip Vision-Exp gate tensors the B12X DSpark draft loader cannot bind."""
- from pathlib import Path
- p = Path("/usr/local/lib/python3.12/dist-packages/vllm/models/deepseek_v4/nvidia/dspark.py")
- text = p.read_text()
- old = ''' if name.endswith(".ffn.gate.bias"):
- name = name.replace(
- ".ffn.gate.bias", ".ffn.gate.e_score_correction_bias"
- )
- param = params_dict[name]
- '''
- new = ''' if name.endswith(".ffn.gate.bias_vl"):
- continue
- if name.endswith(".ffn.gate.bias"):
- name = name.replace(
- ".ffn.gate.bias", ".ffn.gate.e_score_correction_bias"
- )
- if name not in params_dict:
- continue
- param = params_dict[name]
- '''
- if old not in text:
- raise SystemExit(f"patch target missing in {p}")
- p.write_text(text.replace(old, new, 1))
|
