Fix the bug preventing pretrained models to be loaded.
Fix the bug preventing pretrained models to be loaded.

PiperOrigin-RevId: 288938626
Change-Id: I379782cbac91d8fa09d41fd11ab67982a5fed7a7

magenta/ddsp,magenta/ddsp