Hello! I want to apply unsloth_classification notebook for my task. However, I have about 400 classes, which are not represented by single tokens. I believe that the current version of the notebook is unfit for my task. Specifically, the parts with classification head trimming and Data Collator are raising my suspicion, since multi-digit classes are not represented by single tokens. Could you please add another version of unsloth_classification notebook that explaines how to work with classification for 10+ classes? Thank you!