vkc1vk

Results 2 issues of vkc1vk

### System Info N/A ### Who can help? @byshiue ### Information - [X] The official example scripts - [ ] My own modified scripts ### Tasks - [X] An officially...

bug

### 🐛 Describe the bug Tensors saved in `medusa_only_heads` mode are empty. Ref: https://github.com/linkedin/Liger-Kernel/blob/main/examples/medusa/train.py#L392 ### Reproduce _No response_ ### Versions N/A