Instructions to use OzzyGT/YuE2-3B-Diffusers with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use OzzyGT/YuE2-3B-Diffusers with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("OzzyGT/YuE2-3B-Diffusers", 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
Download vae/config.json from OzzyGT/YuE2-3B-Diffusers: direct link, hf CLI and curl.
- Browser
- Download file 313 Bytes
-
https://huggingface.co/OzzyGT/YuE2-3B-Diffusers/resolve/main/vae/config.json
- Command line
-
hf download hf://OzzyGT/YuE2-3B-Diffusers/vae/config.json
-
curl -L -o config.json https://huggingface.co/OzzyGT/YuE2-3B-Diffusers/resolve/main/vae/config.json
313 Bytes
| { | |
| "_class_name": "YuE2VAE", | |
| "encoder_hidden_size": 64, | |
| "decoder_channels": 64, | |
| "decoder_input_channels": 64, | |
| "audio_channels": 2, | |
| "downsampling_ratios": [ | |
| 2, | |
| 2, | |
| 4, | |
| 4, | |
| 5, | |
| 6 | |
| ], | |
| "channel_multiples": [ | |
| 1, | |
| 2, | |
| 4, | |
| 8, | |
| 16, | |
| 32 | |
| ], | |
| "sampling_rate": 48000 | |
| } | |