Use the correct model ID KimiK31ForConditionalGeneration

#1
by akopytko - opened

As per OG's config.json

tiny-random org

Merged. Thanks @akopytko β€” architectures now matches the source config.

yujiepan changed pull request status to merged

Thanks! Also MXFP4 weight_packed tensors in 3D but 2D in the actual thing.

I used this script to fix them: https://gist.github.com/sanmai/e6c737bca4b923b8d36adc708f50362a

Sign up or log in to comment