Some Problem about PRM model

[Alex-Dou123]

Thanks for your good work. I have a questions about coding. I find that for initial PRM (Mistral-7B without finetuning, I have following question.)

obtainscore fail!
Error:'CausalLMOutputWithPast' object has no attribute 'item'

There is no correct values and I find that tensor in "CausalLMOutputWithPast" is a 3D tensor (shape is [1, 2048, 4096]). Therefore, I want to know how I can get the real values?

Thanks