Text-to-Image
Diffusers
UniDiffuserPipeline
image-to-text
image-captioning
image-variation
text-variation
multi-modality
generative model
Instructions to use thu-ml/unidiffuser-v1 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use thu-ml/unidiffuser-v1 with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("thu-ml/unidiffuser-v1", 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
Commit History
Update README.md (#4) 89ee5df
Update README.md (#1) ca0f3d9
Upload 22 files (#3) b78cfee
Update README.md 3dd8509
Update README.md 66b7de3
Update README.md 510eba5
Create README.md 9656653
Delete caption_ae.pt ce7c554
Upload caption_decoder.pth 541e510
uvit_v1 b53ae95
autoencoder_kl, caption_ae af117ed
initial commit 037ba85
fan bao commited on