Upload training_state/optimizer_state.safetensors with huggingface_hub

Upload training_state/optimizer_param_groups.json with huggingface_hub