When set with_color False

Thank you for your awesome work!
I got a question when I use "finetune.sh". When I set with_color=False, my code will report the following problem:
 RuntimeError: shape '[8192, 32, 6]' is invalid for input of size 786432
Can you give me some advices for this error? Thank you for your kindly help!