Instructions to use JJukE/TriplaneVAE with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use JJukE/TriplaneVAE with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("JJukE/TriplaneVAE", 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
Update config.json
Browse files- config.json +1 -2
config.json
CHANGED
|
@@ -14,6 +14,5 @@
|
|
| 14 |
"latent_scale_factor": null,
|
| 15 |
"kl_weight": 1e-06,
|
| 16 |
"geom_weight": 1.0,
|
| 17 |
-
"recon_loss_type": "l2"
|
| 18 |
-
"device": "cuda"
|
| 19 |
}
|
|
|
|
| 14 |
"latent_scale_factor": null,
|
| 15 |
"kl_weight": 1e-06,
|
| 16 |
"geom_weight": 1.0,
|
| 17 |
+
"recon_loss_type": "l2"
|
|
|
|
| 18 |
}
|