swift/plugin/loss_scale/config/alpha_umi.json

{
    "Action:": [2.0, 2.0],
    "Action Input:": [2.0, 2.0],
    "Thought:": [1.0, 1.0],
    "Final Answer:": [1.0, 1.0],
    "Observation:": [2.0, 0.0],
    "Next:": [2,0, 2.0]
}