- [ ] save full model in diffusers format (overrides save only trained parameters) - [ ] when only saving trainable parameters, option to push to tracker - [X] init kwargs for tracker