Humans who prompt or edit the image own the copyright—they're the ones driving the creativity. The AI is just a tool, like Photoshop on steroids.
When an AI image gets sued for infringement, the human takes the legal hit. If you carry all the liability for the output, you should definitely own the rights to it as well.
