Instructions to use Efficient-Large-Model/SANA-WM_bidirectional-diffusers with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use Efficient-Large-Model/SANA-WM_bidirectional-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("Efficient-Large-Model/SANA-WM_bidirectional-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
transformer: diffusers TimestepEmbedding keys, drop unused y_embedding, drop obsolete config keys
Browse files
transformer/config.json
CHANGED
|
@@ -1,13 +1,10 @@
|
|
| 1 |
{
|
| 2 |
"_class_name": "SanaWMTransformer3DModel",
|
| 3 |
"_diffusers_version": "0.39.0.dev0",
|
| 4 |
-
"attn_type": "BidirectionalGDNTriton",
|
| 5 |
"cam_attn_compress": 1,
|
| 6 |
-
"camctrl_type": "BidirectionalGDNUCPESinglePathLiteLABothTriton",
|
| 7 |
"caption_channels": 2304,
|
| 8 |
"chunk_plucker_channels": 48,
|
| 9 |
"chunk_plucker_post_attn_blocks": 20,
|
| 10 |
-
"chunk_split_strategy": "first_chunk_plus_one",
|
| 11 |
"conv_kernel_size": 4,
|
| 12 |
"cross_norm": true,
|
| 13 |
"ffn_type": "GLUMBConvTemp",
|
|
@@ -25,7 +22,6 @@
|
|
| 25 |
null
|
| 26 |
],
|
| 27 |
"mlp_ratio": 3.0,
|
| 28 |
-
"model_max_length": 300,
|
| 29 |
"pos_embed_type": "wan_rope",
|
| 30 |
"pred_sigma": false,
|
| 31 |
"qk_norm": true,
|
|
@@ -33,6 +29,5 @@
|
|
| 33 |
"t_kernel_size": 3,
|
| 34 |
"use_chunk_plucker_post_attn": true,
|
| 35 |
"use_pe": true,
|
| 36 |
-
"y_norm": true
|
| 37 |
-
|
| 38 |
-
}
|
|
|
|
| 1 |
{
|
| 2 |
"_class_name": "SanaWMTransformer3DModel",
|
| 3 |
"_diffusers_version": "0.39.0.dev0",
|
|
|
|
| 4 |
"cam_attn_compress": 1,
|
|
|
|
| 5 |
"caption_channels": 2304,
|
| 6 |
"chunk_plucker_channels": 48,
|
| 7 |
"chunk_plucker_post_attn_blocks": 20,
|
|
|
|
| 8 |
"conv_kernel_size": 4,
|
| 9 |
"cross_norm": true,
|
| 10 |
"ffn_type": "GLUMBConvTemp",
|
|
|
|
| 22 |
null
|
| 23 |
],
|
| 24 |
"mlp_ratio": 3.0,
|
|
|
|
| 25 |
"pos_embed_type": "wan_rope",
|
| 26 |
"pred_sigma": false,
|
| 27 |
"qk_norm": true,
|
|
|
|
| 29 |
"t_kernel_size": 3,
|
| 30 |
"use_chunk_plucker_post_attn": true,
|
| 31 |
"use_pe": true,
|
| 32 |
+
"y_norm": true
|
| 33 |
+
}
|
|
|
transformer/diffusion_pytorch_model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3e1c1c1ced974ea170d6a35687e47fd2ffdd830a68802214b04c55010c47f5b4
|
| 3 |
+
size 5323693080
|