hoping to add multi-language support

#2
by TIN-HF - opened

After some quick tests, English conversation and coding have better performance, but using Chinese for conversation is significantly slower than the baseline.

Red Hat AI org

Hi @TIN-HF , we trained this speculator on Magpie & UltraChat datasets, so we don't expect good performance on other languages, particularly with the reduced vocab likely removing important characters for other languages.

At this stage, we're not targeting multi-language support, but I will point you to the https://github.com/vllm-project/speculators library this was trained with, which you could use to add support yourself.

@fynnsu maybe worth pointing that out in the README.md?

It's ok to decide to support only English, but would be nice to avoid the surprise to other languages users since many would not infer that limitation from the datasets that you mention in the README.md.

Thanks for considering this!!!

Sign up or log in to comment