Add fp16 variants to speedup loading and relax memory constraints (#15)

- Upload folder using huggingface_hub (a6d1f49d89f51ac60ab172accac6dbaec3b34dc3)


Co-authored-by: Sayak Paul <sayakpaul@users.noreply.huggingface.co>

model_index.json

@@ -1,6 +1,7 @@
 {
   "_class_name": "AuraFlowPipeline",
   "_diffusers_version": "0.30.0.dev0",
+  "_name_or_path": "fal/AuraFlow",
   "scheduler": [
     "diffusers",
     "FlowMatchEulerDiscreteScheduler"

text_encoder/config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "EleutherAI/pile-t5-xl",
+  "_name_or_path": "/raid/.cache/huggingface/models--fal--AuraFlow/snapshots/edf69bec4c8c57f5278a655aaca3ceb60d82c0b4/text_encoder",
   "architectures": [
     "UMT5EncoderModel"
   ],
@@ -27,7 +27,7 @@
   "scalable_attention": true,
   "tie_word_embeddings": false,
   "tokenizer_class": "LlamaTokenizerFast",
-  "torch_dtype": "float32",
+  "torch_dtype": "float16",
   "transformers_version": "4.41.2",
   "use_cache": true,
   "vocab_size": 32128

text_encoder/model.fp16.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:decf9b70814ed5e9965bfca9fbd0483462e2bf743790663025b7742f8c014c72
+size 2950448704

tokenizer/special_tokens_map.json

@@ -115,7 +115,13 @@
     "rstrip": false,
     "single_word": false
   },
-  "pad_token": "<s>",
+  "pad_token": {
+    "content": "<s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
   "unk_token": {
     "content": "<unk>",
     "lstrip": false,

transformer/config.json

@@ -1,6 +1,7 @@
 {
   "_class_name": "AuraFlowTransformer2DModel",
   "_diffusers_version": "0.30.0.dev0",
+  "_name_or_path": "/raid/.cache/huggingface/models--fal--AuraFlow/snapshots/edf69bec4c8c57f5278a655aaca3ceb60d82c0b4/transformer",
   "attention_head_dim": 256,
   "caption_projection_dim": 3072,
   "in_channels": 4,

transformer/diffusion_pytorch_model-00001-of-00002.fp16.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:866a485b85f2373604ad6a2f24cc8e57c0e262b88a10a3741af073dd2be9c4cc
+size 9910811936

transformer/diffusion_pytorch_model-00002-of-00002.fp16.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dc714985d7080ea5eaa560f0cffaff4637e18cdec061e3f7b32f42c305ce022c
+size 3774983112

transformer/diffusion_pytorch_model.safetensors.fp16.index.json

@@ -0,0 +1,338 @@
+{
+  "metadata": {
+    "total_size": 13685753856
+  },
+  "weight_map": {
+    "context_embedder.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.0.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.0.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.0.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.0.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.0.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.0.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.0.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.0.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.0.ff.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.0.ff.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.0.ff.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.0.ff_context.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.0.ff_context.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.0.ff_context.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.0.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.0.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.1.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.1.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.1.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.1.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.1.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.1.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.1.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.1.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.1.ff.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.1.ff.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.1.ff.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.1.ff_context.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.1.ff_context.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.1.ff_context.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.1.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.1.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.2.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.2.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.2.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.2.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.2.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.2.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.2.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.2.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.2.ff.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.2.ff.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.2.ff.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.2.ff_context.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.2.ff_context.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.2.ff_context.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.2.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.2.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.3.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.3.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.3.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.3.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.3.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.3.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.3.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.3.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.3.ff.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.3.ff.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.3.ff.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.3.ff_context.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.3.ff_context.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.3.ff_context.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.3.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "joint_transformer_blocks.3.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "norm_out.linear.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "pos_embed.pos_embed": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "pos_embed.proj.bias": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "pos_embed.proj.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "proj_out.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "register_tokens": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.0.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.0.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.0.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.0.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.0.ff.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.0.ff.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.0.ff.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.0.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.1.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.1.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.1.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.1.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.1.ff.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.1.ff.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.1.ff.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.1.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.10.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.10.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.10.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.10.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.10.ff.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.10.ff.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.10.ff.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.10.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.11.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.11.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.11.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.11.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.11.ff.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.11.ff.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.11.ff.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.11.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.12.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.12.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.12.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.12.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.12.ff.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.12.ff.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.12.ff.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.12.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.13.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.13.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.13.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.13.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.13.ff.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.13.ff.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.13.ff.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.13.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.14.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.14.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.14.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.14.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.14.ff.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.14.ff.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.14.ff.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.14.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.15.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.15.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.15.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.15.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.15.ff.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.15.ff.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.15.ff.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.15.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.16.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.16.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.16.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.16.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.16.ff.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.16.ff.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.16.ff.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.16.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.17.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.17.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.17.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.17.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.17.ff.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.17.ff.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.17.ff.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.17.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.18.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.18.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.18.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.18.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.18.ff.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.18.ff.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.18.ff.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.18.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.19.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.19.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.19.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.19.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.19.ff.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.19.ff.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.19.ff.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.19.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.2.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.2.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.2.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.2.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.2.ff.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.2.ff.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.2.ff.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.2.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.20.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.20.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.20.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.20.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.20.ff.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.20.ff.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.20.ff.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.20.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.21.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.21.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.21.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.21.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.21.ff.linear_1.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.21.ff.linear_2.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.21.ff.out_projection.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.21.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.22.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.22.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.22.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.22.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.22.ff.linear_1.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.22.ff.linear_2.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.22.ff.out_projection.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.22.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.23.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.23.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.23.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.23.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.23.ff.linear_1.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.23.ff.linear_2.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.23.ff.out_projection.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.23.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.24.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.24.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.24.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.24.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.24.ff.linear_1.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.24.ff.linear_2.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.24.ff.out_projection.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.24.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.25.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.25.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.25.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.25.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.25.ff.linear_1.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.25.ff.linear_2.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.25.ff.out_projection.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.25.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.26.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.26.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.26.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.26.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.26.ff.linear_1.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.26.ff.linear_2.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.26.ff.out_projection.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.26.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.27.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.27.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.27.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.27.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.27.ff.linear_1.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.27.ff.linear_2.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.27.ff.out_projection.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.27.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.28.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.28.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.28.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.28.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.28.ff.linear_1.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.28.ff.linear_2.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.28.ff.out_projection.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.28.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.29.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.29.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.29.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.29.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.29.ff.linear_1.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.29.ff.linear_2.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.29.ff.out_projection.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.29.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.3.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.3.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.3.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.3.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.3.ff.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.3.ff.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.3.ff.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.3.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.30.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.30.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.30.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.30.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.30.ff.linear_1.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.30.ff.linear_2.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.30.ff.out_projection.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.30.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.31.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.31.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.31.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.31.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.31.ff.linear_1.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.31.ff.linear_2.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.31.ff.out_projection.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.31.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.fp16.safetensors",
+    "single_transformer_blocks.4.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.4.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.4.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.4.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.4.ff.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.4.ff.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.4.ff.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.4.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.5.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.5.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.5.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.5.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.5.ff.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.5.ff.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.5.ff.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.5.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.6.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.6.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.6.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.6.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.6.ff.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.6.ff.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.6.ff.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.6.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.7.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.7.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.7.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.7.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.7.ff.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.7.ff.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.7.ff.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.7.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.8.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.8.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.8.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.8.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.8.ff.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.8.ff.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.8.ff.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.8.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.9.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.9.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.9.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.9.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.9.ff.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.9.ff.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.9.ff.out_projection.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "single_transformer_blocks.9.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "time_step_proj.linear_1.bias": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "time_step_proj.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "time_step_proj.linear_2.bias": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors",
+    "time_step_proj.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.fp16.safetensors"
+  }
+}

vae/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "AutoencoderKL",
   "_diffusers_version": "0.30.0.dev0",
-  "_name_or_path": "stabilityai/sdxl-vae",
+  "_name_or_path": "/raid/.cache/huggingface/models--fal--AuraFlow/snapshots/edf69bec4c8c57f5278a655aaca3ceb60d82c0b4/vae",
   "act_fn": "silu",
   "block_out_channels": [
     128,

vae/diffusion_pytorch_model.fp16.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bcb60880a46b63dea58e9bc591abe15f8350bde47b405f9c38f4be70c6161e68
+size 167335342