from pathlib import Path

import mlx_lm.models


model_path = str(Path(__file__).resolve().parent)
if model_path not in mlx_lm.models.__path__:
    mlx_lm.models.__path__.insert(0, model_path)

from qwen4_cache_integration import register_qwen4_cache_integration
from qwen4_mtp_integration import register_qwen4_mtp_integration


register_qwen4_cache_integration()
register_qwen4_mtp_integration()