How to Load a Fine-Tuned Moirai Model?

[gugasthenpower]

First of all, thank you for this amazing project! I have successfully fine-tuned the Moirai model, and I would like to know the recommended way to load the fine-tuned model for inference.

I tried using torch.load to load the checkpoint, but I encountered some issues related to missing or unexpected keys in the state dictionary. Is there any documentation or example that explains how to properly load and use a fine-tuned Moirai model?

[chenghaoliu89]

Hi @gugasthenpower, you can call uni2ts.model.moirai.MoiraiForecast.load_from_checkpoint(checkpoint_path= ..., ...)

[gugasthenpower]

Thanks for your help! That worked perfectly and helped me a lot.