microsoft

/

mri-autoencoder-v0.1

like 10

Thanks! I think the easiest way to finetune would be to use the diffusers library. The notebook gives an example of how to load the weights for inference. For finetuning you will have to write your own training module. Depending on interest, we will look into open sourcing the training code at some point.