Instructions to use mit-han-lab/nunchaku with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use mit-han-lab/nunchaku with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("mit-han-lab/nunchaku", 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
- Local Apps Settings
- Draw Things
- DiffusionBee
Please consider adding support to torch 2.10.0+cu130
#5
by suffamoph - opened
Beeing using nunchaku models to image making for a while. Thank you!
Please consider adding support to torch 2.12.0+cu132
This is command to setup quickly newest torch version:uv pip install torch==2.12.0+cu132 torchvision==0.27.0+cu132 torchaudio==2.11.0+cu132 --index-url https://download.pytorch.org/whl/cu132 --extra-index-url https://download.pytorch.org/whl/test