Instructions to use gatilin/nitrofusion2-dit-refiners with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use gatilin/nitrofusion2-dit-refiners with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("gatilin/nitrofusion2-dit-refiners", 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
- Local Apps Settings
- Draw Things
- DiffusionBee
Download tensor_mapping.json from gatilin/nitrofusion2-dit-refiners: direct link, hf CLI and curl.
- Browser
- Download file 14.3 kB
-
https://huggingface.co/gatilin/nitrofusion2-dit-refiners/resolve/main/tensor_mapping.json
- Command line
-
hf download hf://gatilin/nitrofusion2-dit-refiners/tensor_mapping.json
-
curl -L -o tensor_mapping.json https://huggingface.co/gatilin/nitrofusion2-dit-refiners/resolve/main/tensor_mapping.json
14.3 kB
| { | |
| "context_refiner.0.attention.norm_k.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 22645888, | |
| 22646144 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 128 | |
| ] | |
| }, | |
| "context_refiner.0.attention.norm_q.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 22646144, | |
| 22646400 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 128 | |
| ] | |
| }, | |
| "context_refiner.0.attention.to_k.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 22646400, | |
| 52137600 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840, | |
| 3840 | |
| ] | |
| }, | |
| "context_refiner.0.attention.to_out.0.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 52137600, | |
| 81628800 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840, | |
| 3840 | |
| ] | |
| }, | |
| "context_refiner.0.attention.to_q.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 81628800, | |
| 111120000 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840, | |
| 3840 | |
| ] | |
| }, | |
| "context_refiner.0.attention.to_v.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 111120000, | |
| 140611200 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840, | |
| 3840 | |
| ] | |
| }, | |
| "context_refiner.0.attention_norm1.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 140611200, | |
| 140618880 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840 | |
| ] | |
| }, | |
| "context_refiner.0.attention_norm2.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 140618880, | |
| 140626560 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840 | |
| ] | |
| }, | |
| "context_refiner.0.feed_forward.w1.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 140626560, | |
| 219269760 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 10240, | |
| 3840 | |
| ] | |
| }, | |
| "context_refiner.0.feed_forward.w2.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 219269760, | |
| 297912960 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840, | |
| 10240 | |
| ] | |
| }, | |
| "context_refiner.0.feed_forward.w3.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 297912960, | |
| 376556160 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 10240, | |
| 3840 | |
| ] | |
| }, | |
| "context_refiner.0.ffn_norm1.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 376556160, | |
| 376563840 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840 | |
| ] | |
| }, | |
| "context_refiner.0.ffn_norm2.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 376563840, | |
| 376571520 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840 | |
| ] | |
| }, | |
| "context_refiner.1.attention.norm_k.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 376571520, | |
| 376571776 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 128 | |
| ] | |
| }, | |
| "context_refiner.1.attention.norm_q.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 376571776, | |
| 376572032 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 128 | |
| ] | |
| }, | |
| "context_refiner.1.attention.to_k.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 376572032, | |
| 406063232 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840, | |
| 3840 | |
| ] | |
| }, | |
| "context_refiner.1.attention.to_out.0.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 406063232, | |
| 435554432 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840, | |
| 3840 | |
| ] | |
| }, | |
| "context_refiner.1.attention.to_q.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 435554432, | |
| 465045632 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840, | |
| 3840 | |
| ] | |
| }, | |
| "context_refiner.1.attention.to_v.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 465045632, | |
| 494536832 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840, | |
| 3840 | |
| ] | |
| }, | |
| "context_refiner.1.attention_norm1.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 494536832, | |
| 494544512 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840 | |
| ] | |
| }, | |
| "context_refiner.1.attention_norm2.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 494544512, | |
| 494552192 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840 | |
| ] | |
| }, | |
| "context_refiner.1.feed_forward.w1.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 494552192, | |
| 573195392 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 10240, | |
| 3840 | |
| ] | |
| }, | |
| "context_refiner.1.feed_forward.w2.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 573195392, | |
| 651838592 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840, | |
| 10240 | |
| ] | |
| }, | |
| "context_refiner.1.feed_forward.w3.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 651838592, | |
| 730481792 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 10240, | |
| 3840 | |
| ] | |
| }, | |
| "context_refiner.1.ffn_norm1.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 730481792, | |
| 730489472 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840 | |
| ] | |
| }, | |
| "context_refiner.1.ffn_norm2.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 730489472, | |
| 730497152 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840 | |
| ] | |
| }, | |
| "noise_refiner.0.adaLN_modulation.0.bias": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4262134912, | |
| 4262165632 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 15360 | |
| ] | |
| }, | |
| "noise_refiner.0.adaLN_modulation.0.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4262165632, | |
| 4270029952 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 15360, | |
| 256 | |
| ] | |
| }, | |
| "noise_refiner.0.attention.norm_k.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4270029952, | |
| 4270030208 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 128 | |
| ] | |
| }, | |
| "noise_refiner.0.attention.norm_q.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4270030208, | |
| 4270030464 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 128 | |
| ] | |
| }, | |
| "noise_refiner.0.attention.to_k.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4270030464, | |
| 4299521664 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840, | |
| 3840 | |
| ] | |
| }, | |
| "noise_refiner.0.attention.to_out.0.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4299521664, | |
| 4329012864 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840, | |
| 3840 | |
| ] | |
| }, | |
| "noise_refiner.0.attention.to_q.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4329012864, | |
| 4358504064 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840, | |
| 3840 | |
| ] | |
| }, | |
| "noise_refiner.0.attention.to_v.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4358504064, | |
| 4387995264 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840, | |
| 3840 | |
| ] | |
| }, | |
| "noise_refiner.0.attention_norm1.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4387995264, | |
| 4388002944 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840 | |
| ] | |
| }, | |
| "noise_refiner.0.attention_norm2.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4388002944, | |
| 4388010624 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840 | |
| ] | |
| }, | |
| "noise_refiner.0.feed_forward.w1.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4388010624, | |
| 4466653824 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 10240, | |
| 3840 | |
| ] | |
| }, | |
| "noise_refiner.0.feed_forward.w2.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4466653824, | |
| 4545297024 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840, | |
| 10240 | |
| ] | |
| }, | |
| "noise_refiner.0.feed_forward.w3.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4545297024, | |
| 4623940224 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 10240, | |
| 3840 | |
| ] | |
| }, | |
| "noise_refiner.0.ffn_norm1.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4623940224, | |
| 4623947904 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840 | |
| ] | |
| }, | |
| "noise_refiner.0.ffn_norm2.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4623947904, | |
| 4623955584 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840 | |
| ] | |
| }, | |
| "noise_refiner.1.adaLN_modulation.0.bias": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4623955584, | |
| 4623986304 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 15360 | |
| ] | |
| }, | |
| "noise_refiner.1.adaLN_modulation.0.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4623986304, | |
| 4631850624 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 15360, | |
| 256 | |
| ] | |
| }, | |
| "noise_refiner.1.attention.norm_k.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4631850624, | |
| 4631850880 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 128 | |
| ] | |
| }, | |
| "noise_refiner.1.attention.norm_q.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4631850880, | |
| 4631851136 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 128 | |
| ] | |
| }, | |
| "noise_refiner.1.attention.to_k.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4631851136, | |
| 4661342336 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840, | |
| 3840 | |
| ] | |
| }, | |
| "noise_refiner.1.attention.to_out.0.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4661342336, | |
| 4690833536 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840, | |
| 3840 | |
| ] | |
| }, | |
| "noise_refiner.1.attention.to_q.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4690833536, | |
| 4720324736 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840, | |
| 3840 | |
| ] | |
| }, | |
| "noise_refiner.1.attention.to_v.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4720324736, | |
| 4749815936 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840, | |
| 3840 | |
| ] | |
| }, | |
| "noise_refiner.1.attention_norm1.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4749815936, | |
| 4749823616 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840 | |
| ] | |
| }, | |
| "noise_refiner.1.attention_norm2.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4749823616, | |
| 4749831296 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840 | |
| ] | |
| }, | |
| "noise_refiner.1.feed_forward.w1.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4749831296, | |
| 4828474496 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 10240, | |
| 3840 | |
| ] | |
| }, | |
| "noise_refiner.1.feed_forward.w2.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4828474496, | |
| 4907117696 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840, | |
| 10240 | |
| ] | |
| }, | |
| "noise_refiner.1.feed_forward.w3.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4907117696, | |
| 4985760896 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 10240, | |
| 3840 | |
| ] | |
| }, | |
| "noise_refiner.1.ffn_norm1.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4985760896, | |
| 4985768576 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840 | |
| ] | |
| }, | |
| "noise_refiner.1.ffn_norm2.weight": { | |
| "source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors", | |
| "source_data_offsets": [ | |
| 4985768576, | |
| 4985776256 | |
| ], | |
| "dtype": "BF16", | |
| "shape": [ | |
| 3840 | |
| ] | |
| } | |
| } | |