Some fine tuning on code generation for this model, result near all is ok
