Update ModelPT level save and restore
# Changelog
- Revert a temporary patch for TLT
- Make ModelPT save_to and restore_from aware of restoration state
- Formalize artifact management

Signed-off-by: smajumdar <titu1994@gmail.com>