Delete auraflow-02

auraflow-02/.gitattributes

@@ -1,37 +0,0 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
-*.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text
-*.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
-*.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.wasm filter=lfs diff=lfs merge=lfs -text
-*.xz filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs -text
-*.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text
-ComfyUI_00381_.webp filter=lfs diff=lfs merge=lfs -text
-comfy[[:space:]]workflow.webp filter=lfs diff=lfs merge=lfs -text

auraflow-02/LICENSE

@@ -1,201 +0,0 @@
-                                 Apache License
-                           Version 2.0, January 2004
-                        http://www.apache.org/licenses/
-
-   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
-
-   1. Definitions.
-
-      "License" shall mean the terms and conditions for use, reproduction,
-      and distribution as defined by Sections 1 through 9 of this document.
-
-      "Licensor" shall mean the copyright owner or entity authorized by
-      the copyright owner that is granting the License.
-
-      "Legal Entity" shall mean the union of the acting entity and all
-      other entities that control, are controlled by, or are under common
-      control with that entity. For the purposes of this definition,
-      "control" means (i) the power, direct or indirect, to cause the
-      direction or management of such entity, whether by contract or
-      otherwise, or (ii) ownership of fifty percent (50%) or more of the
-      outstanding shares, or (iii) beneficial ownership of such entity.
-
-      "You" (or "Your") shall mean an individual or Legal Entity
-      exercising permissions granted by this License.
-
-      "Source" form shall mean the preferred form for making modifications,
-      including but not limited to software source code, documentation
-      source, and configuration files.
-
-      "Object" form shall mean any form resulting from mechanical
-      transformation or translation of a Source form, including but
-      not limited to compiled object code, generated documentation,
-      and conversions to other media types.
-
-      "Work" shall mean the work of authorship, whether in Source or
-      Object form, made available under the License, as indicated by a
-      copyright notice that is included in or attached to the work
-      (an example is provided in the Appendix below).
-
-      "Derivative Works" shall mean any work, whether in Source or Object
-      form, that is based on (or derived from) the Work and for which the
-      editorial revisions, annotations, elaborations, or other modifications
-      represent, as a whole, an original work of authorship. For the purposes
-      of this License, Derivative Works shall not include works that remain
-      separable from, or merely link (or bind by name) to the interfaces of,
-      the Work and Derivative Works thereof.
-
-      "Contribution" shall mean any work of authorship, including
-      the original version of the Work and any modifications or additions
-      to that Work or Derivative Works thereof, that is intentionally
-      submitted to Licensor for inclusion in the Work by the copyright owner
-      or by an individual or Legal Entity authorized to submit on behalf of
-      the copyright owner. For the purposes of this definition, "submitted"
-      means any form of electronic, verbal, or written communication sent
-      to the Licensor or its representatives, including but not limited to
-      communication on electronic mailing lists, source code control systems,
-      and issue tracking systems that are managed by, or on behalf of, the
-      Licensor for the purpose of discussing and improving the Work, but
-      excluding communication that is conspicuously marked or otherwise
-      designated in writing by the copyright owner as "Not a Contribution."
-
-      "Contributor" shall mean Licensor and any individual or Legal Entity
-      on behalf of whom a Contribution has been received by Licensor and
-      subsequently incorporated within the Work.
-
-   2. Grant of Copyright License. Subject to the terms and conditions of
-      this License, each Contributor hereby grants to You a perpetual,
-      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
-      copyright license to reproduce, prepare Derivative Works of,
-      publicly display, publicly perform, sublicense, and distribute the
-      Work and such Derivative Works in Source or Object form.
-
-   3. Grant of Patent License. Subject to the terms and conditions of
-      this License, each Contributor hereby grants to You a perpetual,
-      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
-      (except as stated in this section) patent license to make, have made,
-      use, offer to sell, sell, import, and otherwise transfer the Work,
-      where such license applies only to those patent claims licensable
-      by such Contributor that are necessarily infringed by their
-      Contribution(s) alone or by combination of their Contribution(s)
-      with the Work to which such Contribution(s) was submitted. If You
-      institute patent litigation against any entity (including a
-      cross-claim or counterclaim in a lawsuit) alleging that the Work
-      or a Contribution incorporated within the Work constitutes direct
-      or contributory patent infringement, then any patent licenses
-      granted to You under this License for that Work shall terminate
-      as of the date such litigation is filed.
-
-   4. Redistribution. You may reproduce and distribute copies of the
-      Work or Derivative Works thereof in any medium, with or without
-      modifications, and in Source or Object form, provided that You
-      meet the following conditions:
-
-      (a) You must give any other recipients of the Work or
-          Derivative Works a copy of this License; and
-
-      (b) You must cause any modified files to carry prominent notices
-          stating that You changed the files; and
-
-      (c) You must retain, in the Source form of any Derivative Works
-          that You distribute, all copyright, patent, trademark, and
-          attribution notices from the Source form of the Work,
-          excluding those notices that do not pertain to any part of
-          the Derivative Works; and
-
-      (d) If the Work includes a "NOTICE" text file as part of its
-          distribution, then any Derivative Works that You distribute must
-          include a readable copy of the attribution notices contained
-          within such NOTICE file, excluding those notices that do not
-          pertain to any part of the Derivative Works, in at least one
-          of the following places: within a NOTICE text file distributed
-          as part of the Derivative Works; within the Source form or
-          documentation, if provided along with the Derivative Works; or,
-          within a display generated by the Derivative Works, if and
-          wherever such third-party notices normally appear. The contents
-          of the NOTICE file are for informational purposes only and
-          do not modify the License. You may add Your own attribution
-          notices within Derivative Works that You distribute, alongside
-          or as an addendum to the NOTICE text from the Work, provided
-          that such additional attribution notices cannot be construed
-          as modifying the License.
-
-      You may add Your own copyright statement to Your modifications and
-      may provide additional or different license terms and conditions
-      for use, reproduction, or distribution of Your modifications, or
-      for any such Derivative Works as a whole, provided Your use,
-      reproduction, and distribution of the Work otherwise complies with
-      the conditions stated in this License.
-
-   5. Submission of Contributions. Unless You explicitly state otherwise,
-      any Contribution intentionally submitted for inclusion in the Work
-      by You to the Licensor shall be under the terms and conditions of
-      this License, without any additional terms or conditions.
-      Notwithstanding the above, nothing herein shall supersede or modify
-      the terms of any separate license agreement you may have executed
-      with Licensor regarding such Contributions.
-
-   6. Trademarks. This License does not grant permission to use the trade
-      names, trademarks, service marks, or product names of the Licensor,
-      except as required for reasonable and customary use in describing the
-      origin of the Work and reproducing the content of the NOTICE file.
-
-   7. Disclaimer of Warranty. Unless required by applicable law or
-      agreed to in writing, Licensor provides the Work (and each
-      Contributor provides its Contributions) on an "AS IS" BASIS,
-      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
-      implied, including, without limitation, any warranties or conditions
-      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
-      PARTICULAR PURPOSE. You are solely responsible for determining the
-      appropriateness of using or redistributing the Work and assume any
-      risks associated with Your exercise of permissions under this License.
-
-   8. Limitation of Liability. In no event and under no legal theory,
-      whether in tort (including negligence), contract, or otherwise,
-      unless required by applicable law (such as deliberate and grossly
-      negligent acts) or agreed to in writing, shall any Contributor be
-      liable to You for damages, including any direct, indirect, special,
-      incidental, or consequential damages of any character arising as a
-      result of this License or out of the use or inability to use the
-      Work (including but not limited to damages for loss of goodwill,
-      work stoppage, computer failure or malfunction, or any and all
-      other commercial damages or losses), even if such Contributor
-      has been advised of the possibility of such damages.
-
-   9. Accepting Warranty or Additional Liability. While redistributing
-      the Work or Derivative Works thereof, You may choose to offer,
-      and charge a fee for, acceptance of support, warranty, indemnity,
-      or other liability obligations and/or rights consistent with this
-      License. However, in accepting such obligations, You may act only
-      on Your own behalf and on Your sole responsibility, not on behalf
-      of any other Contributor, and only if You agree to indemnify,
-      defend, and hold each Contributor harmless for any liability
-      incurred by, or claims asserted against, such Contributor by reason
-      of your accepting any such warranty or additional liability.
-
-   END OF TERMS AND CONDITIONS
-
-   APPENDIX: How to apply the Apache License to your work.
-
-      To apply the Apache License to your work, attach the following
-      boilerplate notice, with the fields enclosed by brackets "[]"
-      replaced with your own identifying information. (Don't include
-      the brackets!)  The text should be enclosed in the appropriate
-      comment syntax for the file format. We also recommend that a
-      file or class name and description of purpose be included on the
-      same "printed page" as the copyright notice for easier
-      identification within third-party archives.
-
-   Copyright [yyyy] [name of copyright owner]
-
-   Licensed under the Apache License, Version 2.0 (the "License");
-   you may not use this file except in compliance with the License.
-   You may obtain a copy of the License at
-
-       http://www.apache.org/licenses/LICENSE-2.0
-
-   Unless required by applicable law or agreed to in writing, software
-   distributed under the License is distributed on an "AS IS" BASIS,
-   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-   See the License for the specific language governing permissions and
-   limitations under the License.

auraflow-02/README.md

@@ -1,48 +0,0 @@
----
-license: apache-2.0
-tags:
-- text-to-image
----
-# AuraFlow v0.2
-
-![image/png](https://cdn-uploads.huggingface.co/production/uploads/6380ebb8471a4550ff255c62/jMkLXPFVNGdUb7P4nNTqX.png)
-
-
-AuraFlow v0.2 is the fully open-sourced largest flow-based text-to-image generation model. The model was trained with more compute compared to the previous version, [AuraFlow-v0.1](https://huggingface.co/fal/AuraFlow)
-
-This model achieves state-of-the-art results on GenEval. Read our [blog post](https://blog.fal.ai/auraflow/) for more technical details. You can also check out the comparison with other models on this gallery [page](https://cloneofsimo.github.io/compare_aura_sd3/).
-
-The model is currently in beta. We are working on improving it and the community's feedback is important.
-Join [fal's Discord](https://discord.gg/fal-ai) to give us feedback and stay in touch with the model development.
-
-Credits: A huge thank you to [@cloneofsimo](https://twitter.com/cloneofsimo) and [@isidentical](https://twitter.com/isidentical) for bringing this project to life. It's incredible what two cracked engineers can achieve in
-such a short period of time. We also extend our gratitude to the incredible researchers whose prior work laid the foundation for our efforts.
-
-## Usage
-
-```bash
-$ pip install transformers accelerate protobuf sentencepiece
-$ pip install git+https://github.com/huggingface/diffusers.git
-```
-
-```python
-from diffusers import AuraFlowPipeline
-import torch
-
-pipeline = AuraFlowPipeline.from_pretrained(
-    "fal/AuraFlow-v0.2",
-    torch_dtype=torch.float16,
-    variant="fp16",
-).to("cuda")
-
-image = pipeline(
-    prompt="close-up portrait of a majestic iguana with vibrant blue-green scales, piercing amber eyes, and orange spiky crest. Intricate textures and details visible on scaly skin. Wrapped in dark hood, giving regal appearance. Dramatic lighting against black background. Hyper-realistic, high-resolution image showcasing the reptile's expressive features and coloration.",
-    height=1024,
-    width=1024,
-    num_inference_steps=50, 
-    generator=torch.Generator().manual_seed(666),
-    guidance_scale=3.5,
-).images[0]
-
-image.save("output.png")
-```

auraflow-02/model_index.json

@@ -1,25 +0,0 @@
-{
-  "_class_name": "AuraFlowPipeline",
-  "_diffusers_version": "0.30.0.dev0",
-  "_name_or_path": "fal/AuraFlow-v0.2",
-  "scheduler": [
-    "diffusers",
-    "FlowMatchEulerDiscreteScheduler"
-  ],
-  "text_encoder": [
-    "transformers",
-    "UMT5EncoderModel"
-  ],
-  "tokenizer": [
-    "transformers",
-    "LlamaTokenizerFast"
-  ],
-  "transformer": [
-    "diffusers",
-    "AuraFlowTransformer2DModel"
-  ],
-  "vae": [
-    "diffusers",
-    "AutoencoderKL"
-  ]
-}

auraflow-02/scheduler/scheduler_config.json

@@ -1,6 +0,0 @@
-{
-  "_class_name": "FlowMatchEulerDiscreteScheduler",
-  "_diffusers_version": "0.30.0.dev0",
-  "num_train_timesteps": 1000,
-  "shift": 1.73
-}

auraflow-02/text_encoder/config.json

@@ -1,34 +0,0 @@
-{
-  "_name_or_path": "/raid/.cache/huggingface/models--fal--AuraFlow/snapshots/edf69bec4c8c57f5278a655aaca3ceb60d82c0b4/text_encoder",
-  "architectures": [
-    "UMT5EncoderModel"
-  ],
-  "classifier_dropout": 0.0,
-  "d_ff": 5120,
-  "d_kv": 64,
-  "d_model": 2048,
-  "decoder_start_token_id": 0,
-  "dense_act_fn": "gelu_new",
-  "dropout_rate": 0.1,
-  "eos_token_id": 2,
-  "feed_forward_proj": "gated-gelu",
-  "initializer_factor": 1.0,
-  "is_encoder_decoder": true,
-  "is_gated_act": true,
-  "layer_norm_epsilon": 1e-06,
-  "model_type": "umt5",
-  "num_decoder_layers": 24,
-  "num_heads": 32,
-  "num_layers": 24,
-  "output_past": true,
-  "pad_token_id": 0,
-  "relative_attention_max_distance": 128,
-  "relative_attention_num_buckets": 32,
-  "scalable_attention": true,
-  "tie_word_embeddings": false,
-  "tokenizer_class": "LlamaTokenizerFast",
-  "torch_dtype": "float16",
-  "transformers_version": "4.41.2",
-  "use_cache": true,
-  "vocab_size": 32128
-}

auraflow-02/tokenizer/added_tokens.json

@@ -1,102 +0,0 @@
-{
-  "<extra_id_0>": 32099,
-  "<extra_id_10>": 32089,
-  "<extra_id_11>": 32088,
-  "<extra_id_12>": 32087,
-  "<extra_id_13>": 32086,
-  "<extra_id_14>": 32085,
-  "<extra_id_15>": 32084,
-  "<extra_id_16>": 32083,
-  "<extra_id_17>": 32082,
-  "<extra_id_18>": 32081,
-  "<extra_id_19>": 32080,
-  "<extra_id_1>": 32098,
-  "<extra_id_20>": 32079,
-  "<extra_id_21>": 32078,
-  "<extra_id_22>": 32077,
-  "<extra_id_23>": 32076,
-  "<extra_id_24>": 32075,
-  "<extra_id_25>": 32074,
-  "<extra_id_26>": 32073,
-  "<extra_id_27>": 32072,
-  "<extra_id_28>": 32071,
-  "<extra_id_29>": 32070,
-  "<extra_id_2>": 32097,
-  "<extra_id_30>": 32069,
-  "<extra_id_31>": 32068,
-  "<extra_id_32>": 32067,
-  "<extra_id_33>": 32066,
-  "<extra_id_34>": 32065,
-  "<extra_id_35>": 32064,
-  "<extra_id_36>": 32063,
-  "<extra_id_37>": 32062,
-  "<extra_id_38>": 32061,
-  "<extra_id_39>": 32060,
-  "<extra_id_3>": 32096,
-  "<extra_id_40>": 32059,
-  "<extra_id_41>": 32058,
-  "<extra_id_42>": 32057,
-  "<extra_id_43>": 32056,
-  "<extra_id_44>": 32055,
-  "<extra_id_45>": 32054,
-  "<extra_id_46>": 32053,
-  "<extra_id_47>": 32052,
-  "<extra_id_48>": 32051,
-  "<extra_id_49>": 32050,
-  "<extra_id_4>": 32095,
-  "<extra_id_50>": 32049,
-  "<extra_id_51>": 32048,
-  "<extra_id_52>": 32047,
-  "<extra_id_53>": 32046,
-  "<extra_id_54>": 32045,
-  "<extra_id_55>": 32044,
-  "<extra_id_56>": 32043,
-  "<extra_id_57>": 32042,
-  "<extra_id_58>": 32041,
-  "<extra_id_59>": 32040,
-  "<extra_id_5>": 32094,
-  "<extra_id_60>": 32039,
-  "<extra_id_61>": 32038,
-  "<extra_id_62>": 32037,
-  "<extra_id_63>": 32036,
-  "<extra_id_64>": 32035,
-  "<extra_id_65>": 32034,
-  "<extra_id_66>": 32033,
-  "<extra_id_67>": 32032,
-  "<extra_id_68>": 32031,
-  "<extra_id_69>": 32030,
-  "<extra_id_6>": 32093,
-  "<extra_id_70>": 32029,
-  "<extra_id_71>": 32028,
-  "<extra_id_72>": 32027,
-  "<extra_id_73>": 32026,
-  "<extra_id_74>": 32025,
-  "<extra_id_75>": 32024,
-  "<extra_id_76>": 32023,
-  "<extra_id_77>": 32022,
-  "<extra_id_78>": 32021,
-  "<extra_id_79>": 32020,
-  "<extra_id_7>": 32092,
-  "<extra_id_80>": 32019,
-  "<extra_id_81>": 32018,
-  "<extra_id_82>": 32017,
-  "<extra_id_83>": 32016,
-  "<extra_id_84>": 32015,
-  "<extra_id_85>": 32014,
-  "<extra_id_86>": 32013,
-  "<extra_id_87>": 32012,
-  "<extra_id_88>": 32011,
-  "<extra_id_89>": 32010,
-  "<extra_id_8>": 32091,
-  "<extra_id_90>": 32009,
-  "<extra_id_91>": 32008,
-  "<extra_id_92>": 32007,
-  "<extra_id_93>": 32006,
-  "<extra_id_94>": 32005,
-  "<extra_id_95>": 32004,
-  "<extra_id_96>": 32003,
-  "<extra_id_97>": 32002,
-  "<extra_id_98>": 32001,
-  "<extra_id_99>": 32000,
-  "<extra_id_9>": 32090
-}

auraflow-02/tokenizer/special_tokens_map.json

@@ -1,132 +0,0 @@
-{
-  "additional_special_tokens": [
-    "<extra_id_99>",
-    "<extra_id_98>",
-    "<extra_id_97>",
-    "<extra_id_96>",
-    "<extra_id_95>",
-    "<extra_id_94>",
-    "<extra_id_93>",
-    "<extra_id_92>",
-    "<extra_id_91>",
-    "<extra_id_90>",
-    "<extra_id_89>",
-    "<extra_id_88>",
-    "<extra_id_87>",
-    "<extra_id_86>",
-    "<extra_id_85>",
-    "<extra_id_84>",
-    "<extra_id_83>",
-    "<extra_id_82>",
-    "<extra_id_81>",
-    "<extra_id_80>",
-    "<extra_id_79>",
-    "<extra_id_78>",
-    "<extra_id_77>",
-    "<extra_id_76>",
-    "<extra_id_75>",
-    "<extra_id_74>",
-    "<extra_id_73>",
-    "<extra_id_72>",
-    "<extra_id_71>",
-    "<extra_id_70>",
-    "<extra_id_69>",
-    "<extra_id_68>",
-    "<extra_id_67>",
-    "<extra_id_66>",
-    "<extra_id_65>",
-    "<extra_id_64>",
-    "<extra_id_63>",
-    "<extra_id_62>",
-    "<extra_id_61>",
-    "<extra_id_60>",
-    "<extra_id_59>",
-    "<extra_id_58>",
-    "<extra_id_57>",
-    "<extra_id_56>",
-    "<extra_id_55>",
-    "<extra_id_54>",
-    "<extra_id_53>",
-    "<extra_id_52>",
-    "<extra_id_51>",
-    "<extra_id_50>",
-    "<extra_id_49>",
-    "<extra_id_48>",
-    "<extra_id_47>",
-    "<extra_id_46>",
-    "<extra_id_45>",
-    "<extra_id_44>",
-    "<extra_id_43>",
-    "<extra_id_42>",
-    "<extra_id_41>",
-    "<extra_id_40>",
-    "<extra_id_39>",
-    "<extra_id_38>",
-    "<extra_id_37>",
-    "<extra_id_36>",
-    "<extra_id_35>",
-    "<extra_id_34>",
-    "<extra_id_33>",
-    "<extra_id_32>",
-    "<extra_id_31>",
-    "<extra_id_30>",
-    "<extra_id_29>",
-    "<extra_id_28>",
-    "<extra_id_27>",
-    "<extra_id_26>",
-    "<extra_id_25>",
-    "<extra_id_24>",
-    "<extra_id_23>",
-    "<extra_id_22>",
-    "<extra_id_21>",
-    "<extra_id_20>",
-    "<extra_id_19>",
-    "<extra_id_18>",
-    "<extra_id_17>",
-    "<extra_id_16>",
-    "<extra_id_15>",
-    "<extra_id_14>",
-    "<extra_id_13>",
-    "<extra_id_12>",
-    "<extra_id_11>",
-    "<extra_id_10>",
-    "<extra_id_9>",
-    "<extra_id_8>",
-    "<extra_id_7>",
-    "<extra_id_6>",
-    "<extra_id_5>",
-    "<extra_id_4>",
-    "<extra_id_3>",
-    "<extra_id_2>",
-    "<extra_id_1>",
-    "<extra_id_0>"
-  ],
-  "bos_token": {
-    "content": "<s>",
-    "lstrip": false,
-    "normalized": false,
-    "rstrip": false,
-    "single_word": false
-  },
-  "eos_token": {
-    "content": "</s>",
-    "lstrip": false,
-    "normalized": false,
-    "rstrip": false,
-    "single_word": false
-  },
-  "pad_token": {
-    "content": "<s>",
-    "lstrip": false,
-    "normalized": false,
-    "rstrip": false,
-    "single_word": false
-  },
-  "unk_token": {
-    "content": "<unk>",
-    "lstrip": false,
-    "normalized": false,
-    "rstrip": false,
-    "single_word": false
-  }
-}

auraflow-02/tokenizer/tokenizer.model

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

auraflow-02/tokenizer/tokenizer_config.json

@@ -1,945 +0,0 @@
-{
-  "add_bos_token": false,
-  "add_eos_token": true,
-  "add_prefix_space": true,
-  "added_tokens_decoder": {
-    "0": {
-      "content": "<unk>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "1": {
-      "content": "<s>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "2": {
-      "content": "</s>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32000": {
-      "content": "<extra_id_99>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32001": {
-      "content": "<extra_id_98>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32002": {
-      "content": "<extra_id_97>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32003": {
-      "content": "<extra_id_96>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32004": {
-      "content": "<extra_id_95>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32005": {
-      "content": "<extra_id_94>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32006": {
-      "content": "<extra_id_93>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32007": {
-      "content": "<extra_id_92>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32008": {
-      "content": "<extra_id_91>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32009": {
-      "content": "<extra_id_90>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32010": {
-      "content": "<extra_id_89>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32011": {
-      "content": "<extra_id_88>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32012": {
-      "content": "<extra_id_87>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32013": {
-      "content": "<extra_id_86>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32014": {
-      "content": "<extra_id_85>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32015": {
-      "content": "<extra_id_84>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32016": {
-      "content": "<extra_id_83>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32017": {
-      "content": "<extra_id_82>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32018": {
-      "content": "<extra_id_81>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32019": {
-      "content": "<extra_id_80>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32020": {
-      "content": "<extra_id_79>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32021": {
-      "content": "<extra_id_78>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32022": {
-      "content": "<extra_id_77>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32023": {
-      "content": "<extra_id_76>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32024": {
-      "content": "<extra_id_75>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32025": {
-      "content": "<extra_id_74>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32026": {
-      "content": "<extra_id_73>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32027": {
-      "content": "<extra_id_72>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32028": {
-      "content": "<extra_id_71>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32029": {
-      "content": "<extra_id_70>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32030": {
-      "content": "<extra_id_69>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32031": {
-      "content": "<extra_id_68>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32032": {
-      "content": "<extra_id_67>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32033": {
-      "content": "<extra_id_66>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32034": {
-      "content": "<extra_id_65>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32035": {
-      "content": "<extra_id_64>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32036": {
-      "content": "<extra_id_63>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32037": {
-      "content": "<extra_id_62>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32038": {
-      "content": "<extra_id_61>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32039": {
-      "content": "<extra_id_60>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32040": {
-      "content": "<extra_id_59>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32041": {
-      "content": "<extra_id_58>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32042": {
-      "content": "<extra_id_57>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32043": {
-      "content": "<extra_id_56>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32044": {
-      "content": "<extra_id_55>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32045": {
-      "content": "<extra_id_54>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32046": {
-      "content": "<extra_id_53>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32047": {
-      "content": "<extra_id_52>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32048": {
-      "content": "<extra_id_51>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32049": {
-      "content": "<extra_id_50>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32050": {
-      "content": "<extra_id_49>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32051": {
-      "content": "<extra_id_48>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32052": {
-      "content": "<extra_id_47>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32053": {
-      "content": "<extra_id_46>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32054": {
-      "content": "<extra_id_45>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32055": {
-      "content": "<extra_id_44>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32056": {
-      "content": "<extra_id_43>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32057": {
-      "content": "<extra_id_42>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32058": {
-      "content": "<extra_id_41>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32059": {
-      "content": "<extra_id_40>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32060": {
-      "content": "<extra_id_39>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32061": {
-      "content": "<extra_id_38>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32062": {
-      "content": "<extra_id_37>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32063": {
-      "content": "<extra_id_36>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32064": {
-      "content": "<extra_id_35>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32065": {
-      "content": "<extra_id_34>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32066": {
-      "content": "<extra_id_33>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32067": {
-      "content": "<extra_id_32>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32068": {
-      "content": "<extra_id_31>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32069": {
-      "content": "<extra_id_30>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32070": {
-      "content": "<extra_id_29>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32071": {
-      "content": "<extra_id_28>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32072": {
-      "content": "<extra_id_27>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32073": {
-      "content": "<extra_id_26>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32074": {
-      "content": "<extra_id_25>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32075": {
-      "content": "<extra_id_24>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32076": {
-      "content": "<extra_id_23>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32077": {
-      "content": "<extra_id_22>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32078": {
-      "content": "<extra_id_21>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32079": {
-      "content": "<extra_id_20>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32080": {
-      "content": "<extra_id_19>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32081": {
-      "content": "<extra_id_18>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32082": {
-      "content": "<extra_id_17>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32083": {
-      "content": "<extra_id_16>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32084": {
-      "content": "<extra_id_15>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32085": {
-      "content": "<extra_id_14>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32086": {
-      "content": "<extra_id_13>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32087": {
-      "content": "<extra_id_12>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32088": {
-      "content": "<extra_id_11>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32089": {
-      "content": "<extra_id_10>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32090": {
-      "content": "<extra_id_9>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32091": {
-      "content": "<extra_id_8>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32092": {
-      "content": "<extra_id_7>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32093": {
-      "content": "<extra_id_6>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32094": {
-      "content": "<extra_id_5>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32095": {
-      "content": "<extra_id_4>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32096": {
-      "content": "<extra_id_3>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32097": {
-      "content": "<extra_id_2>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32098": {
-      "content": "<extra_id_1>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "32099": {
-      "content": "<extra_id_0>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    }
-  },
-  "additional_special_tokens": [
-    "<extra_id_99>",
-    "<extra_id_98>",
-    "<extra_id_97>",
-    "<extra_id_96>",
-    "<extra_id_95>",
-    "<extra_id_94>",
-    "<extra_id_93>",
-    "<extra_id_92>",
-    "<extra_id_91>",
-    "<extra_id_90>",
-    "<extra_id_89>",
-    "<extra_id_88>",
-    "<extra_id_87>",
-    "<extra_id_86>",
-    "<extra_id_85>",
-    "<extra_id_84>",
-    "<extra_id_83>",
-    "<extra_id_82>",
-    "<extra_id_81>",
-    "<extra_id_80>",
-    "<extra_id_79>",
-    "<extra_id_78>",
-    "<extra_id_77>",
-    "<extra_id_76>",
-    "<extra_id_75>",
-    "<extra_id_74>",
-    "<extra_id_73>",
-    "<extra_id_72>",
-    "<extra_id_71>",
-    "<extra_id_70>",
-    "<extra_id_69>",
-    "<extra_id_68>",
-    "<extra_id_67>",
-    "<extra_id_66>",
-    "<extra_id_65>",
-    "<extra_id_64>",
-    "<extra_id_63>",
-    "<extra_id_62>",
-    "<extra_id_61>",
-    "<extra_id_60>",
-    "<extra_id_59>",
-    "<extra_id_58>",
-    "<extra_id_57>",
-    "<extra_id_56>",
-    "<extra_id_55>",
-    "<extra_id_54>",
-    "<extra_id_53>",
-    "<extra_id_52>",
-    "<extra_id_51>",
-    "<extra_id_50>",
-    "<extra_id_49>",
-    "<extra_id_48>",
-    "<extra_id_47>",
-    "<extra_id_46>",
-    "<extra_id_45>",
-    "<extra_id_44>",
-    "<extra_id_43>",
-    "<extra_id_42>",
-    "<extra_id_41>",
-    "<extra_id_40>",
-    "<extra_id_39>",
-    "<extra_id_38>",
-    "<extra_id_37>",
-    "<extra_id_36>",
-    "<extra_id_35>",
-    "<extra_id_34>",
-    "<extra_id_33>",
-    "<extra_id_32>",
-    "<extra_id_31>",
-    "<extra_id_30>",
-    "<extra_id_29>",
-    "<extra_id_28>",
-    "<extra_id_27>",
-    "<extra_id_26>",
-    "<extra_id_25>",
-    "<extra_id_24>",
-    "<extra_id_23>",
-    "<extra_id_22>",
-    "<extra_id_21>",
-    "<extra_id_20>",
-    "<extra_id_19>",
-    "<extra_id_18>",
-    "<extra_id_17>",
-    "<extra_id_16>",
-    "<extra_id_15>",
-    "<extra_id_14>",
-    "<extra_id_13>",
-    "<extra_id_12>",
-    "<extra_id_11>",
-    "<extra_id_10>",
-    "<extra_id_9>",
-    "<extra_id_8>",
-    "<extra_id_7>",
-    "<extra_id_6>",
-    "<extra_id_5>",
-    "<extra_id_4>",
-    "<extra_id_3>",
-    "<extra_id_2>",
-    "<extra_id_1>",
-    "<extra_id_0>"
-  ],
-  "bos_token": "<s>",
-  "clean_up_tokenization_spaces": false,
-  "eos_token": "</s>",
-  "legacy": false,
-  "model_max_length": 512,
-  "pad_token": "<s>",
-  "padding_side": "right",
-  "sp_model_kwargs": {},
-  "spaces_between_special_tokens": false,
-  "tokenizer_class": "LlamaTokenizer",
-  "unk_token": "<unk>",
-  "use_default_system_prompt": false
-}

auraflow-02/transformer/config.json

@@ -1,15 +0,0 @@
-{
-  "_class_name": "AuraFlowTransformer2DModel",
-  "_diffusers_version": "0.30.0.dev0",
-  "attention_head_dim": 256,
-  "caption_projection_dim": 3072,
-  "in_channels": 4,
-  "joint_attention_dim": 2048,
-  "num_attention_heads": 12,
-  "num_mmdit_layers": 4,
-  "num_single_dit_layers": 32,
-  "out_channels": 4,
-  "patch_size": 2,
-  "pos_embed_max_size": 4096,
-  "sample_size": 64
-}

auraflow-02/transformer/diffusion_pytorch_model.safetensors.fp16.index.json

@@ -1,338 +0,0 @@
-{
-  "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"
-  }
-}

auraflow-02/transformer/diffusion_pytorch_model.safetensors.index.json

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

auraflow-02/vae/config.json

@@ -1,37 +0,0 @@
-{
-  "_class_name": "AutoencoderKL",
-  "_diffusers_version": "0.30.0.dev0",
-  "_name_or_path": "/raid/.cache/huggingface/models--fal--AuraFlow/snapshots/edf69bec4c8c57f5278a655aaca3ceb60d82c0b4/vae",
-  "act_fn": "silu",
-  "block_out_channels": [
-    128,
-    256,
-    512,
-    512
-  ],
-  "down_block_types": [
-    "DownEncoderBlock2D",
-    "DownEncoderBlock2D",
-    "DownEncoderBlock2D",
-    "DownEncoderBlock2D"
-  ],
-  "force_upcast": true,
-  "in_channels": 3,
-  "latent_channels": 4,
-  "latents_mean": null,
-  "latents_std": null,
-  "layers_per_block": 2,
-  "norm_num_groups": 32,
-  "out_channels": 3,
-  "sample_size": 1024,
-  "scaling_factor": 0.13025,
-  "shift_factor": null,
-  "up_block_types": [
-    "UpDecoderBlock2D",
-    "UpDecoderBlock2D",
-    "UpDecoderBlock2D",
-    "UpDecoderBlock2D"
-  ],
-  "use_post_quant_conv": true,
-  "use_quant_conv": true
-}