Is this for just like auto complete, because you are not going to get anything very useful out of a code-only training set.
So i needed to make fundamental arquitecture changes .Do some KV cache tricks.
And then prove the new arquitecture was faster with benchmarks and perplexity was acceptable.