How to use customized trained models?

[ScottishFold007]

I trained a vocos model with a model I trained myself, how do I use it? I see that the existing model loaded is in pytorch_model.bin format, and the parameters in it are named differently than the directly trained one? Can you give me an example of how to use it, thanks!

[patriotyk]

def load_vocos(checkpoint_path, config_path, device):
    model = Vocos.from_hparams(config_path).to(device)

    raw_model = torch.load(checkpoint_path, map_location=device)
    raw_model = raw_model if 'state_dict' not in raw_model else raw_model['state_dict']
    model.load_state_dict(raw_model, strict=False)
    model.eval()
    return model