GMM: Cannot load a checkpointed UBM

Trying to load a previously checkpointed UBM fails with ValueError: A UBM is required for MAP GMM.