Rename trainer_state.json to config_a/trainer_state.json

Rename model.safetensors to config_a/model.safetensors

Rename config.json to config_a/config.json