Another option,
Remove the “disable input” node and add branches to your input axis.
Player presses “run attack” that sets a bool “RunAttackPressed” ,
From input axis set a float “ForwardInput” then have a branch using “Run Attack Pressed” to allow input to continue, play a constant input based on the stored float value or completely disable as per existing setup.
Yuo can use animnotifiers to reset the bool, or release key press, what ever.
This way with simple branches you can decide if movement is constant, effected by player input or stopped.
This would also allow you to decide if turning is allowed by adding same to turn axis input.
and also allows disable root motion, might be handy for collision etc.
Later you could replace branches and bools with enums and switches, depending on how complicated you need it to be.