1from svara.codec.mimi import MimiCodec
2import torch
3
4codec = MimiCodec.from_pretrained("kyutai/mimi", dtype=torch.bfloat16)
5state = torch.load("mimi_final.pt", map_location="cpu")
6codec.model.load_state_dict(state, strict=False)mimi_final.pt — final checkpoint (full model state_dict)mimi_step*.pt — intermediate checkpoints