[float8] keep model.output as nn.Linear (high precision, not fp8)

[float8] keep model.output as nn.Linear (high precision, not fp8) #1169