Add configurable timeout to initialization procedure

Fix bug in await initialization timeout. If the last message in the sync topic has a stale rebalance generation ID, it's possible that we never update the state from START_INITIAL_SYNC, which would cause awaitInitialization to timeout.