def METHOD_NAME(
    cls,
    additional_dir: Workspace,
    workspace: Workspace,
    task: Task,
):
    all_states = dict(
        Config={},  # placeholder
        SpecHead=task.predictor.state_dict(),
        Transformer=task.upstream.state_dict(),
        Upstream_Config=dict(
            transformer=_transformer_config,
            audio=_audio_config,
            task=dict(sequence_length=0),
        ),
    )
    torch.save(
        all_states, str(additional_dir.parent.resolve()) + "/all_states.ckpt"
    )