Instructions to use stabilityai/sd-vae-ft-mse with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use stabilityai/sd-vae-ft-mse with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("stabilityai/sd-vae-ft-mse", 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
Add pipeline tag, links to Github and project page
#14
by nielsr HF Staff - opened
This PR ensures it can be found at https://huggingface.co/models?pipeline_tag=image-to-image. It also adds a link to the paper page, as well as a link to the Github and project page.