Edit button does not stop output generation. When you press edit, you still need to wait until output is fully done before you can edit anything. And as I said, this is too much work to edit and ask AI to continue every time until I see passable result. That's why I said, that it would be great for QoL