Avoid saving deepspeed optimizer and model states at checkpoints

It looks really hard…