Fix proto conversion (which fixes other issues like breakage in model inference tests).
Fix proto conversion (which fixes other issues like breakage in model inference tests).

PiperOrigin-RevId: 376954601
Change-Id: I61aa7cc5a515cf14a27189d04ae82e3b7ea1bebd

google-research/falken,google-research/falken,google-research/falken,google-research/falken,google-research/falken,google-research/falken,google-research/falken