Ryan Taylor
Ryan Taylor
Unfortunately, the fix you introduced assumes that calling `.forward_()` produces a valid result when called on the original input. What needs to happen during inference is for `.forward()` to use...