Add loading ckpt logic from state_dict and not directly from model class

Add ControlNet and Scheduler