where is the loss code?

[royshan]

First, I'd like to thank you very much for this inspiring work!

In your paper, you mentioned "For Baize v1.5, following Vicuna (Chiang et al., 2023), we only compute loss for AI’s responses in the dialogue transcript." I am curious about your implementation and may customize it. I searched the repo but couldn't find it. Could you give me a pointer for that?

[albertwy]

Maybe the version of this code is too old