Instructions to use microsoft/mri-autoencoder-v0.1 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use microsoft/mri-autoencoder-v0.1 with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("microsoft/mri-autoencoder-v0.1", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
Training Set.
#1
by niaouris - opened
Hello,
What is the training set for this autoencoder?
Did you use only the training set from FastMRI, or did you train it on validation and test sets too?
Thanks in advance.
Sorry for the delayed response. Yes, I used only the training set for the multicoil knee and brain datasets.