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 transformer/config.json from OzzyGT/YuE2-3B-Diffusers: direct link, hf CLI and curl.
- Browser
- Download file 388 Bytes
-
https://huggingface.co/OzzyGT/YuE2-3B-Diffusers/resolve/main/transformer/config.json
- Command line
-
hf download hf://OzzyGT/YuE2-3B-Diffusers/transformer/config.json
-
curl -L -o config.json https://huggingface.co/OzzyGT/YuE2-3B-Diffusers/resolve/main/transformer/config.json
388 Bytes
| { | |
| "_class_name": "YuE2TransformerModel", | |
| "hidden_size": 2048, | |
| "num_layers": 28, | |
| "num_attention_heads": 16, | |
| "num_key_value_heads": 8, | |
| "attention_head_dim": 128, | |
| "intermediate_size": 6144, | |
| "vocab_size": 184704, | |
| "norm_eps": 1e-06, | |
| "rope_theta": 1000000, | |
| "max_position_embeddings": 24576, | |
| "latent_channels": 64, | |
| "max_latent_frames": 24576, | |
| "timestep_shift": 1.0 | |
| } | |