Instructions to use davidberenstein1957/Sana_600M_512px_diffusers with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Pruna AI
How to use davidberenstein1957/Sana_600M_512px_diffusers with Pruna AI:
from pruna import PrunaModel pip install -U diffusers transformers accelerate
from pruna import PrunaModel import torch # switch to "mps" for apple devices pipe = PrunaModel.from_pretrained("davidberenstein1957/Sana_600M_512px_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
| { | |
| "_class_name": "SanaTransformer2DModel", | |
| "_diffusers_version": "0.34.0.dev0", | |
| "_name_or_path": "/Users/davidberenstein/.cache/huggingface/hub/models--Efficient-Large-Model--Sana_600M_512px_diffusers/snapshots/2defc07f5fb66d0c53ace051585e9a2cb83f8c15/transformer", | |
| "attention_bias": false, | |
| "attention_head_dim": 32, | |
| "caption_channels": 2304, | |
| "cross_attention_dim": 1152, | |
| "cross_attention_head_dim": 72, | |
| "dropout": 0.0, | |
| "guidance_embeds": false, | |
| "guidance_embeds_scale": 0.1, | |
| "in_channels": 32, | |
| "interpolation_scale": null, | |
| "mlp_ratio": 2.5, | |
| "norm_elementwise_affine": false, | |
| "norm_eps": 1e-06, | |
| "num_attention_heads": 36, | |
| "num_cross_attention_heads": 16, | |
| "num_layers": 28, | |
| "out_channels": 32, | |
| "patch_size": 1, | |
| "qk_norm": null, | |
| "sample_size": 16, | |
| "timestep_scale": 1.0 | |
| } | |