Remove unnecessary sharding logic for configuring Orbax single replica restore and broadcast.#3598
Open
copybara-service[bot] wants to merge 1 commit intomainfrom
Open
Remove unnecessary sharding logic for configuring Orbax single replica restore and broadcast.#3598copybara-service[bot] wants to merge 1 commit intomainfrom
copybara-service[bot] wants to merge 1 commit intomainfrom
