Fix build of model builder test after interface change.
Fix build of model builder test after interface change.

PiperOrigin-RevId: 317415534

google/iree,google/iree,google/iree,iree-org/iree,iree-org/iree,google/iree,google/iree,iree-org/iree,iree-org/iree,iree-org/iree,google/iree,iree-org/iree,google/iree,iree-org/iree