Save model weight only if require gradient updates.

[dxli94]

For partial model training, e.g. BLIP2, it is not necessary to checkpoint the entire model. Saving partial weights is sufficient.

This is currently not supported by the runner.