SAN FRANCISCO/TOKYO, June 9 (Reuters) - Private equity firm Bain Capital is replacing rival KKR & Co LP in a Japanese government-led consortium that also includes Western Digital Corp to bid for Toshiba Corp's chip unit, sources familiar with the matter said.