def METHOD_NAME(self):
    config = {
        "num_tokens": self.num_tokens,
        "output_mode": self.output_mode,
    }
    base_config = super().METHOD_NAME()
    return {**base_config, **config}