Closed stephantul closed 1 month ago
We didn't move the tensor to cpu before trying to dump it, which causes an error if the model is on another device.
All modified and coverable lines are covered by tests :white_check_mark:
96.74% <ø> (ø)
We didn't move the tensor to cpu before trying to dump it, which causes an error if the model is on another device.