You can either revance your yt music app, or use Firefox with an extension that keeps videos playing while your phone is locked.