Upload 19 files

LICENSE

@@ -0,0 +1,201 @@
+                                 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.

README.md

@@ -0,0 +1,24 @@
+---
+language:
+- en
+- zh
+license: apache-2.0
+base_model:
+  - Qwen/Qwen-Image-2512
+base_model_relation: quantized
+tags:
+- image-generation
+- qwen-image
+- sdnq
+pipeline_tag: text-to-image
+library_name: diffusers
+---
+# Qwen Image 2512 SDNQ Dynamic INT4
+
+This is an int4 quantized version of [Qwen/Qwen-Image-2512](https://huggingface.co/Qwen/Qwen-Image-2512) using [SDNQ](https://github.com/Disty0/sdnq) (SD.Next Quantization) with the dynamic option and Hadamard Rotation.
+
+Note: You need SDNQ v0.2.0 or superior
+
+## Usage
+
+You can find ready-to-use scripts in the [diffusers-recipes](https://github.com/asomoza/diffusers-recipes/blob/main/models/qwen_image_2512/README.md) repository.

model_index.json

@@ -0,0 +1,24 @@
+{
+  "_class_name": "QwenImagePipeline",
+  "_diffusers_version": "0.36.0.dev0",
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "text_encoder": [
+    "transformers",
+    "Qwen2_5_VLForConditionalGeneration"
+  ],
+  "tokenizer": [
+    "transformers",
+    "Qwen2Tokenizer"
+  ],
+  "transformer": [
+    "diffusers",
+    "QwenImageTransformer2DModel"
+  ],
+  "vae": [
+    "diffusers",
+    "AutoencoderKLQwenImage"
+  ]
+}

scheduler/scheduler_config.json

@@ -0,0 +1,18 @@
+{
+  "_class_name": "FlowMatchEulerDiscreteScheduler",
+  "_diffusers_version": "0.36.0.dev0",
+  "base_image_seq_len": 256,
+  "base_shift": 0.5,
+  "invert_sigmas": false,
+  "max_image_seq_len": 8192,
+  "max_shift": 0.9,
+  "num_train_timesteps": 1000,
+  "shift": 1.0,
+  "shift_terminal": 0.02,
+  "stochastic_sampling": false,
+  "time_shift_type": "exponential",
+  "use_beta_sigmas": false,
+  "use_dynamic_shifting": true,
+  "use_exponential_sigmas": false,
+  "use_karras_sigmas": false
+}

text_encoder/config.json

@@ -0,0 +1,674 @@
+{
+  "architectures": [
+    "Qwen2_5_VLForConditionalGeneration"
+  ],
+  "attention_dropout": 0.0,
+  "bos_token_id": 151643,
+  "dtype": "bfloat16",
+  "eos_token_id": 151645,
+  "hidden_act": "silu",
+  "hidden_size": 3584,
+  "image_token_id": 151655,
+  "initializer_range": 0.02,
+  "intermediate_size": 18944,
+  "max_position_embeddings": 128000,
+  "max_window_layers": 28,
+  "model_type": "qwen2_5_vl",
+  "num_attention_heads": 28,
+  "num_hidden_layers": 28,
+  "num_key_value_heads": 4,
+  "quantization_config": {
+    "add_skip_keys": true,
+    "dequantize_fp32": false,
+    "dynamic_loss_threshold": 0.01,
+    "group_size": 0,
+    "hadamard_group_size": 256,
+    "is_integer": true,
+    "is_training": false,
+    "modules_dtype_dict": {
+      "uint4": [
+        "model.language_model.layers.0.mlp.down_proj.weight",
+        "model.language_model.layers.0.mlp.gate_proj.weight",
+        "model.language_model.layers.0.mlp.up_proj.weight",
+        "model.language_model.layers.0.self_attn.k_proj.weight",
+        "model.language_model.layers.0.self_attn.o_proj.weight",
+        "model.language_model.layers.0.self_attn.q_proj.weight",
+        "model.language_model.layers.0.self_attn.v_proj.weight",
+        "model.language_model.layers.1.mlp.down_proj.weight",
+        "model.language_model.layers.1.mlp.gate_proj.weight",
+        "model.language_model.layers.1.mlp.up_proj.weight",
+        "model.language_model.layers.1.self_attn.k_proj.weight",
+        "model.language_model.layers.1.self_attn.o_proj.weight",
+        "model.language_model.layers.1.self_attn.q_proj.weight",
+        "model.language_model.layers.1.self_attn.v_proj.weight",
+        "model.language_model.layers.2.mlp.down_proj.weight",
+        "model.language_model.layers.2.mlp.gate_proj.weight",
+        "model.language_model.layers.2.mlp.up_proj.weight",
+        "model.language_model.layers.2.self_attn.k_proj.weight",
+        "model.language_model.layers.2.self_attn.o_proj.weight",
+        "model.language_model.layers.2.self_attn.q_proj.weight",
+        "model.language_model.layers.2.self_attn.v_proj.weight",
+        "model.language_model.layers.3.mlp.down_proj.weight",
+        "model.language_model.layers.3.mlp.gate_proj.weight",
+        "model.language_model.layers.3.mlp.up_proj.weight",
+        "model.language_model.layers.3.self_attn.k_proj.weight",
+        "model.language_model.layers.3.self_attn.o_proj.weight",
+        "model.language_model.layers.3.self_attn.q_proj.weight",
+        "model.language_model.layers.3.self_attn.v_proj.weight",
+        "model.language_model.layers.4.mlp.down_proj.weight",
+        "model.language_model.layers.4.mlp.gate_proj.weight",
+        "model.language_model.layers.4.mlp.up_proj.weight",
+        "model.language_model.layers.4.self_attn.k_proj.weight",
+        "model.language_model.layers.4.self_attn.o_proj.weight",
+        "model.language_model.layers.4.self_attn.q_proj.weight",
+        "model.language_model.layers.4.self_attn.v_proj.weight",
+        "model.language_model.layers.5.mlp.down_proj.weight",
+        "model.language_model.layers.5.mlp.gate_proj.weight",
+        "model.language_model.layers.5.mlp.up_proj.weight",
+        "model.language_model.layers.5.self_attn.k_proj.weight",
+        "model.language_model.layers.5.self_attn.o_proj.weight",
+        "model.language_model.layers.5.self_attn.q_proj.weight",
+        "model.language_model.layers.5.self_attn.v_proj.weight",
+        "model.language_model.layers.6.mlp.down_proj.weight",
+        "model.language_model.layers.6.mlp.gate_proj.weight",
+        "model.language_model.layers.6.mlp.up_proj.weight",
+        "model.language_model.layers.6.self_attn.k_proj.weight",
+        "model.language_model.layers.6.self_attn.o_proj.weight",
+        "model.language_model.layers.6.self_attn.q_proj.weight",
+        "model.language_model.layers.6.self_attn.v_proj.weight",
+        "model.language_model.layers.7.mlp.down_proj.weight",
+        "model.language_model.layers.7.mlp.gate_proj.weight",
+        "model.language_model.layers.7.mlp.up_proj.weight",
+        "model.language_model.layers.7.self_attn.k_proj.weight",
+        "model.language_model.layers.7.self_attn.o_proj.weight",
+        "model.language_model.layers.7.self_attn.q_proj.weight",
+        "model.language_model.layers.7.self_attn.v_proj.weight",
+        "model.language_model.layers.8.mlp.down_proj.weight",
+        "model.language_model.layers.8.mlp.gate_proj.weight",
+        "model.language_model.layers.8.mlp.up_proj.weight",
+        "model.language_model.layers.8.self_attn.k_proj.weight",
+        "model.language_model.layers.8.self_attn.o_proj.weight",
+        "model.language_model.layers.8.self_attn.q_proj.weight",
+        "model.language_model.layers.8.self_attn.v_proj.weight",
+        "model.language_model.layers.9.mlp.down_proj.weight",
+        "model.language_model.layers.9.mlp.gate_proj.weight",
+        "model.language_model.layers.9.mlp.up_proj.weight",
+        "model.language_model.layers.9.self_attn.k_proj.weight",
+        "model.language_model.layers.9.self_attn.o_proj.weight",
+        "model.language_model.layers.9.self_attn.q_proj.weight",
+        "model.language_model.layers.9.self_attn.v_proj.weight",
+        "model.language_model.layers.10.mlp.down_proj.weight",
+        "model.language_model.layers.10.mlp.gate_proj.weight",
+        "model.language_model.layers.10.mlp.up_proj.weight",
+        "model.language_model.layers.10.self_attn.k_proj.weight",
+        "model.language_model.layers.10.self_attn.o_proj.weight",
+        "model.language_model.layers.10.self_attn.q_proj.weight",
+        "model.language_model.layers.10.self_attn.v_proj.weight",
+        "model.language_model.layers.11.mlp.down_proj.weight",
+        "model.language_model.layers.11.mlp.gate_proj.weight",
+        "model.language_model.layers.11.mlp.up_proj.weight",
+        "model.language_model.layers.11.self_attn.k_proj.weight",
+        "model.language_model.layers.11.self_attn.o_proj.weight",
+        "model.language_model.layers.11.self_attn.q_proj.weight",
+        "model.language_model.layers.11.self_attn.v_proj.weight",
+        "model.language_model.layers.12.mlp.down_proj.weight",
+        "model.language_model.layers.12.mlp.gate_proj.weight",
+        "model.language_model.layers.12.mlp.up_proj.weight",
+        "model.language_model.layers.12.self_attn.k_proj.weight",
+        "model.language_model.layers.12.self_attn.o_proj.weight",
+        "model.language_model.layers.12.self_attn.q_proj.weight",
+        "model.language_model.layers.12.self_attn.v_proj.weight",
+        "model.language_model.layers.13.mlp.down_proj.weight",
+        "model.language_model.layers.13.mlp.gate_proj.weight",
+        "model.language_model.layers.13.mlp.up_proj.weight",
+        "model.language_model.layers.13.self_attn.k_proj.weight",
+        "model.language_model.layers.13.self_attn.o_proj.weight",
+        "model.language_model.layers.13.self_attn.q_proj.weight",
+        "model.language_model.layers.13.self_attn.v_proj.weight",
+        "model.language_model.layers.14.mlp.down_proj.weight",
+        "model.language_model.layers.14.mlp.gate_proj.weight",
+        "model.language_model.layers.14.mlp.up_proj.weight",
+        "model.language_model.layers.14.self_attn.k_proj.weight",
+        "model.language_model.layers.14.self_attn.o_proj.weight",
+        "model.language_model.layers.14.self_attn.q_proj.weight",
+        "model.language_model.layers.14.self_attn.v_proj.weight",
+        "model.language_model.layers.15.mlp.down_proj.weight",
+        "model.language_model.layers.15.mlp.gate_proj.weight",
+        "model.language_model.layers.15.mlp.up_proj.weight",
+        "model.language_model.layers.15.self_attn.k_proj.weight",
+        "model.language_model.layers.15.self_attn.o_proj.weight",
+        "model.language_model.layers.15.self_attn.q_proj.weight",
+        "model.language_model.layers.15.self_attn.v_proj.weight",
+        "model.language_model.layers.16.mlp.down_proj.weight",
+        "model.language_model.layers.16.mlp.gate_proj.weight",
+        "model.language_model.layers.16.mlp.up_proj.weight",
+        "model.language_model.layers.16.self_attn.k_proj.weight",
+        "model.language_model.layers.16.self_attn.o_proj.weight",
+        "model.language_model.layers.16.self_attn.q_proj.weight",
+        "model.language_model.layers.16.self_attn.v_proj.weight",
+        "model.language_model.layers.17.mlp.down_proj.weight",
+        "model.language_model.layers.17.mlp.gate_proj.weight",
+        "model.language_model.layers.17.mlp.up_proj.weight",
+        "model.language_model.layers.17.self_attn.k_proj.weight",
+        "model.language_model.layers.17.self_attn.o_proj.weight",
+        "model.language_model.layers.17.self_attn.q_proj.weight",
+        "model.language_model.layers.17.self_attn.v_proj.weight",
+        "model.language_model.layers.18.mlp.down_proj.weight",
+        "model.language_model.layers.18.mlp.gate_proj.weight",
+        "model.language_model.layers.18.mlp.up_proj.weight",
+        "model.language_model.layers.18.self_attn.k_proj.weight",
+        "model.language_model.layers.18.self_attn.o_proj.weight",
+        "model.language_model.layers.18.self_attn.q_proj.weight",
+        "model.language_model.layers.18.self_attn.v_proj.weight",
+        "model.language_model.layers.19.mlp.down_proj.weight",
+        "model.language_model.layers.19.mlp.gate_proj.weight",
+        "model.language_model.layers.19.mlp.up_proj.weight",
+        "model.language_model.layers.19.self_attn.k_proj.weight",
+        "model.language_model.layers.19.self_attn.o_proj.weight",
+        "model.language_model.layers.19.self_attn.q_proj.weight",
+        "model.language_model.layers.19.self_attn.v_proj.weight",
+        "model.language_model.layers.20.mlp.down_proj.weight",
+        "model.language_model.layers.20.mlp.gate_proj.weight",
+        "model.language_model.layers.20.mlp.up_proj.weight",
+        "model.language_model.layers.20.self_attn.k_proj.weight",
+        "model.language_model.layers.20.self_attn.o_proj.weight",
+        "model.language_model.layers.20.self_attn.q_proj.weight",
+        "model.language_model.layers.20.self_attn.v_proj.weight",
+        "model.language_model.layers.21.mlp.down_proj.weight",
+        "model.language_model.layers.21.mlp.gate_proj.weight",
+        "model.language_model.layers.21.mlp.up_proj.weight",
+        "model.language_model.layers.21.self_attn.k_proj.weight",
+        "model.language_model.layers.21.self_attn.o_proj.weight",
+        "model.language_model.layers.21.self_attn.q_proj.weight",
+        "model.language_model.layers.21.self_attn.v_proj.weight",
+        "model.language_model.layers.22.mlp.down_proj.weight",
+        "model.language_model.layers.22.mlp.gate_proj.weight",
+        "model.language_model.layers.22.mlp.up_proj.weight",
+        "model.language_model.layers.22.self_attn.k_proj.weight",
+        "model.language_model.layers.22.self_attn.o_proj.weight",
+        "model.language_model.layers.22.self_attn.q_proj.weight",
+        "model.language_model.layers.22.self_attn.v_proj.weight",
+        "model.language_model.layers.23.mlp.down_proj.weight",
+        "model.language_model.layers.23.mlp.gate_proj.weight",
+        "model.language_model.layers.23.mlp.up_proj.weight",
+        "model.language_model.layers.23.self_attn.k_proj.weight",
+        "model.language_model.layers.23.self_attn.o_proj.weight",
+        "model.language_model.layers.23.self_attn.q_proj.weight",
+        "model.language_model.layers.23.self_attn.v_proj.weight",
+        "model.language_model.layers.24.mlp.down_proj.weight",
+        "model.language_model.layers.24.mlp.gate_proj.weight",
+        "model.language_model.layers.24.mlp.up_proj.weight",
+        "model.language_model.layers.24.self_attn.k_proj.weight",
+        "model.language_model.layers.24.self_attn.o_proj.weight",
+        "model.language_model.layers.24.self_attn.q_proj.weight",
+        "model.language_model.layers.24.self_attn.v_proj.weight",
+        "model.language_model.layers.25.mlp.down_proj.weight",
+        "model.language_model.layers.25.mlp.gate_proj.weight",
+        "model.language_model.layers.25.mlp.up_proj.weight",
+        "model.language_model.layers.25.self_attn.k_proj.weight",
+        "model.language_model.layers.25.self_attn.o_proj.weight",
+        "model.language_model.layers.25.self_attn.q_proj.weight",
+        "model.language_model.layers.25.self_attn.v_proj.weight",
+        "model.language_model.layers.26.mlp.down_proj.weight",
+        "model.language_model.layers.26.mlp.gate_proj.weight",
+        "model.language_model.layers.26.mlp.up_proj.weight",
+        "model.language_model.layers.26.self_attn.k_proj.weight",
+        "model.language_model.layers.26.self_attn.o_proj.weight",
+        "model.language_model.layers.26.self_attn.q_proj.weight",
+        "model.language_model.layers.26.self_attn.v_proj.weight",
+        "model.language_model.layers.27.mlp.down_proj.weight",
+        "model.language_model.layers.27.mlp.gate_proj.weight",
+        "model.language_model.layers.27.mlp.up_proj.weight",
+        "model.language_model.layers.27.self_attn.k_proj.weight",
+        "model.language_model.layers.27.self_attn.o_proj.weight",
+        "model.language_model.layers.27.self_attn.q_proj.weight",
+        "model.language_model.layers.27.self_attn.v_proj.weight",
+        "model.visual.blocks.0.attn.proj.weight",
+        "model.visual.blocks.0.attn.qkv.weight",
+        "model.visual.blocks.0.mlp.down_proj.weight",
+        "model.visual.blocks.0.mlp.gate_proj.weight",
+        "model.visual.blocks.0.mlp.up_proj.weight",
+        "model.visual.blocks.1.attn.proj.weight",
+        "model.visual.blocks.1.attn.qkv.weight",
+        "model.visual.blocks.1.mlp.down_proj.weight",
+        "model.visual.blocks.1.mlp.gate_proj.weight",
+        "model.visual.blocks.1.mlp.up_proj.weight",
+        "model.visual.blocks.2.attn.proj.weight",
+        "model.visual.blocks.2.attn.qkv.weight",
+        "model.visual.blocks.2.mlp.down_proj.weight",
+        "model.visual.blocks.2.mlp.gate_proj.weight",
+        "model.visual.blocks.2.mlp.up_proj.weight",
+        "model.visual.blocks.3.attn.proj.weight",
+        "model.visual.blocks.3.attn.qkv.weight",
+        "model.visual.blocks.3.mlp.down_proj.weight",
+        "model.visual.blocks.3.mlp.gate_proj.weight",
+        "model.visual.blocks.3.mlp.up_proj.weight",
+        "model.visual.blocks.4.attn.proj.weight",
+        "model.visual.blocks.4.attn.qkv.weight",
+        "model.visual.blocks.4.mlp.down_proj.weight",
+        "model.visual.blocks.4.mlp.gate_proj.weight",
+        "model.visual.blocks.4.mlp.up_proj.weight",
+        "model.visual.blocks.5.attn.proj.weight",
+        "model.visual.blocks.5.attn.qkv.weight",
+        "model.visual.blocks.5.mlp.down_proj.weight",
+        "model.visual.blocks.5.mlp.gate_proj.weight",
+        "model.visual.blocks.5.mlp.up_proj.weight",
+        "model.visual.blocks.6.attn.proj.weight",
+        "model.visual.blocks.6.attn.qkv.weight",
+        "model.visual.blocks.6.mlp.down_proj.weight",
+        "model.visual.blocks.6.mlp.gate_proj.weight",
+        "model.visual.blocks.6.mlp.up_proj.weight",
+        "model.visual.blocks.7.attn.proj.weight",
+        "model.visual.blocks.7.attn.qkv.weight",
+        "model.visual.blocks.7.mlp.down_proj.weight",
+        "model.visual.blocks.7.mlp.gate_proj.weight",
+        "model.visual.blocks.7.mlp.up_proj.weight",
+        "model.visual.blocks.8.attn.proj.weight",
+        "model.visual.blocks.8.attn.qkv.weight",
+        "model.visual.blocks.8.mlp.down_proj.weight",
+        "model.visual.blocks.8.mlp.gate_proj.weight",
+        "model.visual.blocks.8.mlp.up_proj.weight",
+        "model.visual.blocks.9.attn.proj.weight",
+        "model.visual.blocks.9.attn.qkv.weight",
+        "model.visual.blocks.9.mlp.down_proj.weight",
+        "model.visual.blocks.9.mlp.gate_proj.weight",
+        "model.visual.blocks.9.mlp.up_proj.weight",
+        "model.visual.blocks.10.attn.proj.weight",
+        "model.visual.blocks.10.attn.qkv.weight",
+        "model.visual.blocks.10.mlp.down_proj.weight",
+        "model.visual.blocks.10.mlp.gate_proj.weight",
+        "model.visual.blocks.10.mlp.up_proj.weight",
+        "model.visual.blocks.11.attn.proj.weight",
+        "model.visual.blocks.11.attn.qkv.weight",
+        "model.visual.blocks.11.mlp.down_proj.weight",
+        "model.visual.blocks.11.mlp.gate_proj.weight",
+        "model.visual.blocks.11.mlp.up_proj.weight",
+        "model.visual.blocks.12.attn.proj.weight",
+        "model.visual.blocks.12.attn.qkv.weight",
+        "model.visual.blocks.12.mlp.down_proj.weight",
+        "model.visual.blocks.12.mlp.gate_proj.weight",
+        "model.visual.blocks.12.mlp.up_proj.weight",
+        "model.visual.blocks.13.attn.proj.weight",
+        "model.visual.blocks.13.attn.qkv.weight",
+        "model.visual.blocks.13.mlp.down_proj.weight",
+        "model.visual.blocks.13.mlp.gate_proj.weight",
+        "model.visual.blocks.13.mlp.up_proj.weight",
+        "model.visual.blocks.14.attn.proj.weight",
+        "model.visual.blocks.14.attn.qkv.weight",
+        "model.visual.blocks.14.mlp.down_proj.weight",
+        "model.visual.blocks.14.mlp.gate_proj.weight",
+        "model.visual.blocks.14.mlp.up_proj.weight",
+        "model.visual.blocks.15.attn.proj.weight",
+        "model.visual.blocks.15.attn.qkv.weight",
+        "model.visual.blocks.15.mlp.down_proj.weight",
+        "model.visual.blocks.15.mlp.gate_proj.weight",
+        "model.visual.blocks.15.mlp.up_proj.weight",
+        "model.visual.blocks.16.attn.proj.weight",
+        "model.visual.blocks.16.attn.qkv.weight",
+        "model.visual.blocks.16.mlp.down_proj.weight",
+        "model.visual.blocks.16.mlp.gate_proj.weight",
+        "model.visual.blocks.16.mlp.up_proj.weight",
+        "model.visual.blocks.17.attn.proj.weight",
+        "model.visual.blocks.17.attn.qkv.weight",
+        "model.visual.blocks.17.mlp.down_proj.weight",
+        "model.visual.blocks.17.mlp.gate_proj.weight",
+        "model.visual.blocks.17.mlp.up_proj.weight",
+        "model.visual.blocks.18.attn.proj.weight",
+        "model.visual.blocks.18.attn.qkv.weight",
+        "model.visual.blocks.18.mlp.down_proj.weight",
+        "model.visual.blocks.18.mlp.gate_proj.weight",
+        "model.visual.blocks.18.mlp.up_proj.weight",
+        "model.visual.blocks.19.attn.proj.weight",
+        "model.visual.blocks.19.attn.qkv.weight",
+        "model.visual.blocks.19.mlp.down_proj.weight",
+        "model.visual.blocks.19.mlp.gate_proj.weight",
+        "model.visual.blocks.19.mlp.up_proj.weight",
+        "model.visual.blocks.20.attn.proj.weight",
+        "model.visual.blocks.20.attn.qkv.weight",
+        "model.visual.blocks.20.mlp.down_proj.weight",
+        "model.visual.blocks.20.mlp.gate_proj.weight",
+        "model.visual.blocks.20.mlp.up_proj.weight",
+        "model.visual.blocks.21.attn.proj.weight",
+        "model.visual.blocks.21.attn.qkv.weight",
+        "model.visual.blocks.21.mlp.down_proj.weight",
+        "model.visual.blocks.21.mlp.gate_proj.weight",
+        "model.visual.blocks.21.mlp.up_proj.weight",
+        "model.visual.blocks.22.attn.proj.weight",
+        "model.visual.blocks.22.attn.qkv.weight",
+        "model.visual.blocks.22.mlp.down_proj.weight",
+        "model.visual.blocks.22.mlp.gate_proj.weight",
+        "model.visual.blocks.22.mlp.up_proj.weight",
+        "model.visual.blocks.23.attn.proj.weight",
+        "model.visual.blocks.23.attn.qkv.weight",
+        "model.visual.blocks.23.mlp.down_proj.weight",
+        "model.visual.blocks.23.mlp.gate_proj.weight",
+        "model.visual.blocks.23.mlp.up_proj.weight",
+        "model.visual.blocks.24.attn.proj.weight",
+        "model.visual.blocks.24.attn.qkv.weight",
+        "model.visual.blocks.24.mlp.down_proj.weight",
+        "model.visual.blocks.24.mlp.gate_proj.weight",
+        "model.visual.blocks.24.mlp.up_proj.weight",
+        "model.visual.blocks.25.attn.proj.weight",
+        "model.visual.blocks.25.attn.qkv.weight",
+        "model.visual.blocks.25.mlp.down_proj.weight",
+        "model.visual.blocks.25.mlp.gate_proj.weight",
+        "model.visual.blocks.25.mlp.up_proj.weight",
+        "model.visual.blocks.26.attn.proj.weight",
+        "model.visual.blocks.26.attn.qkv.weight",
+        "model.visual.blocks.26.mlp.down_proj.weight",
+        "model.visual.blocks.26.mlp.gate_proj.weight",
+        "model.visual.blocks.26.mlp.up_proj.weight",
+        "model.visual.blocks.27.attn.proj.weight",
+        "model.visual.blocks.27.attn.qkv.weight",
+        "model.visual.blocks.27.mlp.down_proj.weight",
+        "model.visual.blocks.27.mlp.gate_proj.weight",
+        "model.visual.blocks.27.mlp.up_proj.weight",
+        "model.visual.blocks.28.attn.proj.weight",
+        "model.visual.blocks.28.attn.qkv.weight",
+        "model.visual.blocks.28.mlp.down_proj.weight",
+        "model.visual.blocks.28.mlp.gate_proj.weight",
+        "model.visual.blocks.28.mlp.up_proj.weight",
+        "model.visual.blocks.29.attn.proj.weight",
+        "model.visual.blocks.29.attn.qkv.weight",
+        "model.visual.blocks.29.mlp.down_proj.weight",
+        "model.visual.blocks.29.mlp.gate_proj.weight",
+        "model.visual.blocks.29.mlp.up_proj.weight",
+        "model.visual.blocks.30.attn.proj.weight",
+        "model.visual.blocks.30.attn.qkv.weight",
+        "model.visual.blocks.30.mlp.down_proj.weight",
+        "model.visual.blocks.30.mlp.gate_proj.weight",
+        "model.visual.blocks.30.mlp.up_proj.weight",
+        "model.visual.blocks.31.attn.proj.weight",
+        "model.visual.blocks.31.attn.qkv.weight",
+        "model.visual.blocks.31.mlp.down_proj.weight",
+        "model.visual.blocks.31.mlp.gate_proj.weight",
+        "model.visual.blocks.31.mlp.up_proj.weight",
+        "model.visual.merger.mlp.0.weight",
+        "model.visual.merger.mlp.2.weight"
+      ]
+    },
+    "modules_quant_config": {},
+    "modules_to_not_convert": [
+      "patch_embedding",
+      ".time_embed",
+      "multi_modal_projector",
+      ".vid_out",
+      "time_text_embed",
+      ".emb_in",
+      ".proj_out",
+      ".vid_in",
+      ".context_embedder",
+      ".txt_in",
+      ".final_layer",
+      ".condition_embedder",
+      ".txt_out",
+      "patch_emb",
+      ".img_in",
+      ".norm_out",
+      "lm_head",
+      ".emb_out",
+      ".t_embedder",
+      ".img_out",
+      "wte",
+      ".y_embedder",
+      "lm_head.weight",
+      "patch_embed",
+      ".x_embedder",
+      "model.language_model.embed_tokens.weight",
+      "model.language_model.layers.0.input_layernorm.weight",
+      "model.language_model.layers.0.post_attention_layernorm.weight",
+      "model.language_model.layers.1.input_layernorm.weight",
+      "model.language_model.layers.1.post_attention_layernorm.weight",
+      "model.language_model.layers.2.input_layernorm.weight",
+      "model.language_model.layers.2.post_attention_layernorm.weight",
+      "model.language_model.layers.3.input_layernorm.weight",
+      "model.language_model.layers.3.post_attention_layernorm.weight",
+      "model.language_model.layers.4.input_layernorm.weight",
+      "model.language_model.layers.4.post_attention_layernorm.weight",
+      "model.language_model.layers.5.input_layernorm.weight",
+      "model.language_model.layers.5.post_attention_layernorm.weight",
+      "model.language_model.layers.6.input_layernorm.weight",
+      "model.language_model.layers.6.post_attention_layernorm.weight",
+      "model.language_model.layers.7.input_layernorm.weight",
+      "model.language_model.layers.7.post_attention_layernorm.weight",
+      "model.language_model.layers.8.input_layernorm.weight",
+      "model.language_model.layers.8.post_attention_layernorm.weight",
+      "model.language_model.layers.9.input_layernorm.weight",
+      "model.language_model.layers.9.post_attention_layernorm.weight",
+      "model.language_model.layers.10.input_layernorm.weight",
+      "model.language_model.layers.10.post_attention_layernorm.weight",
+      "model.language_model.layers.11.input_layernorm.weight",
+      "model.language_model.layers.11.post_attention_layernorm.weight",
+      "model.language_model.layers.12.input_layernorm.weight",
+      "model.language_model.layers.12.post_attention_layernorm.weight",
+      "model.language_model.layers.13.input_layernorm.weight",
+      "model.language_model.layers.13.post_attention_layernorm.weight",
+      "model.language_model.layers.14.input_layernorm.weight",
+      "model.language_model.layers.14.post_attention_layernorm.weight",
+      "model.language_model.layers.15.input_layernorm.weight",
+      "model.language_model.layers.15.post_attention_layernorm.weight",
+      "model.language_model.layers.16.input_layernorm.weight",
+      "model.language_model.layers.16.post_attention_layernorm.weight",
+      "model.language_model.layers.17.input_layernorm.weight",
+      "model.language_model.layers.17.post_attention_layernorm.weight",
+      "model.language_model.layers.18.input_layernorm.weight",
+      "model.language_model.layers.18.post_attention_layernorm.weight",
+      "model.language_model.layers.19.input_layernorm.weight",
+      "model.language_model.layers.19.post_attention_layernorm.weight",
+      "model.language_model.layers.20.input_layernorm.weight",
+      "model.language_model.layers.20.post_attention_layernorm.weight",
+      "model.language_model.layers.21.input_layernorm.weight",
+      "model.language_model.layers.21.post_attention_layernorm.weight",
+      "model.language_model.layers.22.input_layernorm.weight",
+      "model.language_model.layers.22.post_attention_layernorm.weight",
+      "model.language_model.layers.23.input_layernorm.weight",
+      "model.language_model.layers.23.post_attention_layernorm.weight",
+      "model.language_model.layers.24.input_layernorm.weight",
+      "model.language_model.layers.24.post_attention_layernorm.weight",
+      "model.language_model.layers.25.input_layernorm.weight",
+      "model.language_model.layers.25.post_attention_layernorm.weight",
+      "model.language_model.layers.26.input_layernorm.weight",
+      "model.language_model.layers.26.post_attention_layernorm.weight",
+      "model.language_model.layers.27.input_layernorm.weight",
+      "model.language_model.layers.27.post_attention_layernorm.weight",
+      "model.language_model.norm.weight",
+      "model.visual.blocks.0.norm1.weight",
+      "model.visual.blocks.0.norm2.weight",
+      "model.visual.blocks.1.norm1.weight",
+      "model.visual.blocks.1.norm2.weight",
+      "model.visual.blocks.2.norm1.weight",
+      "model.visual.blocks.2.norm2.weight",
+      "model.visual.blocks.3.norm1.weight",
+      "model.visual.blocks.3.norm2.weight",
+      "model.visual.blocks.4.norm1.weight",
+      "model.visual.blocks.4.norm2.weight",
+      "model.visual.blocks.5.norm1.weight",
+      "model.visual.blocks.5.norm2.weight",
+      "model.visual.blocks.6.norm1.weight",
+      "model.visual.blocks.6.norm2.weight",
+      "model.visual.blocks.7.norm1.weight",
+      "model.visual.blocks.7.norm2.weight",
+      "model.visual.blocks.8.norm1.weight",
+      "model.visual.blocks.8.norm2.weight",
+      "model.visual.blocks.9.norm1.weight",
+      "model.visual.blocks.9.norm2.weight",
+      "model.visual.blocks.10.norm1.weight",
+      "model.visual.blocks.10.norm2.weight",
+      "model.visual.blocks.11.norm1.weight",
+      "model.visual.blocks.11.norm2.weight",
+      "model.visual.blocks.12.norm1.weight",
+      "model.visual.blocks.12.norm2.weight",
+      "model.visual.blocks.13.norm1.weight",
+      "model.visual.blocks.13.norm2.weight",
+      "model.visual.blocks.14.norm1.weight",
+      "model.visual.blocks.14.norm2.weight",
+      "model.visual.blocks.15.norm1.weight",
+      "model.visual.blocks.15.norm2.weight",
+      "model.visual.blocks.16.norm1.weight",
+      "model.visual.blocks.16.norm2.weight",
+      "model.visual.blocks.17.norm1.weight",
+      "model.visual.blocks.17.norm2.weight",
+      "model.visual.blocks.18.norm1.weight",
+      "model.visual.blocks.18.norm2.weight",
+      "model.visual.blocks.19.norm1.weight",
+      "model.visual.blocks.19.norm2.weight",
+      "model.visual.blocks.20.norm1.weight",
+      "model.visual.blocks.20.norm2.weight",
+      "model.visual.blocks.21.norm1.weight",
+      "model.visual.blocks.21.norm2.weight",
+      "model.visual.blocks.22.norm1.weight",
+      "model.visual.blocks.22.norm2.weight",
+      "model.visual.blocks.23.norm1.weight",
+      "model.visual.blocks.23.norm2.weight",
+      "model.visual.blocks.24.norm1.weight",
+      "model.visual.blocks.24.norm2.weight",
+      "model.visual.blocks.25.norm1.weight",
+      "model.visual.blocks.25.norm2.weight",
+      "model.visual.blocks.26.norm1.weight",
+      "model.visual.blocks.26.norm2.weight",
+      "model.visual.blocks.27.norm1.weight",
+      "model.visual.blocks.27.norm2.weight",
+      "model.visual.blocks.28.norm1.weight",
+      "model.visual.blocks.28.norm2.weight",
+      "model.visual.blocks.29.norm1.weight",
+      "model.visual.blocks.29.norm2.weight",
+      "model.visual.blocks.30.norm1.weight",
+      "model.visual.blocks.30.norm2.weight",
+      "model.visual.blocks.31.norm1.weight",
+      "model.visual.blocks.31.norm2.weight",
+      "model.visual.merger.ln_q.weight"
+    ],
+    "modules_to_not_use_matmul": [],
+    "non_blocking": false,
+    "quant_conv": false,
+    "quant_embedding": false,
+    "quant_method": "sdnq",
+    "quantization_device": null,
+    "quantized_matmul_dtype": null,
+    "return_device": null,
+    "sdnq_version": "0.2.0",
+    "svd_rank": 32,
+    "svd_steps": 8,
+    "use_dynamic_quantization": true,
+    "use_grad_ckpt": true,
+    "use_hadamard": true,
+    "use_quantized_matmul": false,
+    "use_quantized_matmul_conv": false,
+    "use_static_quantization": true,
+    "use_stochastic_rounding": false,
+    "use_svd": false,
+    "weights_dtype": "uint4"
+  },
+  "rms_norm_eps": 1e-06,
+  "rope_parameters": {
+    "mrope_section": [
+      16,
+      24,
+      24
+    ],
+    "rope_type": "default",
+    "type": "default"
+  },
+  "rope_theta": 1000000.0,
+  "sliding_window": 32768,
+  "text_config": {
+    "_name_or_path": "/cpfs01/haoyangzhang/pretrained_weights/Qwen2.5-VL",
+    "architectures": [
+      "Qwen2_5_VLForConditionalGeneration"
+    ],
+    "attention_dropout": 0.0,
+    "bos_token_id": 151643,
+    "dtype": "bfloat16",
+    "eos_token_id": 151645,
+    "hidden_act": "silu",
+    "hidden_size": 3584,
+    "initializer_range": 0.02,
+    "intermediate_size": 18944,
+    "layer_types": [
+      "full_attention",
+      "full_attention",
+      "full_attention",
+      "full_attention",
+      "full_attention",
+      "full_attention",
+      "full_attention",
+      "full_attention",
+      "full_attention",
+      "full_attention",
+      "full_attention",
+      "full_attention",
+      "full_attention",
+      "full_attention",
+      "full_attention",
+      "full_attention",
+      "full_attention",
+      "full_attention",
+      "full_attention",
+      "full_attention",
+      "full_attention",
+      "full_attention",
+      "full_attention",
+      "full_attention",
+      "full_attention",
+      "full_attention",
+      "full_attention",
+      "full_attention"
+    ],
+    "max_position_embeddings": 128000,
+    "max_window_layers": 28,
+    "model_type": "qwen2_5_vl_text",
+    "num_attention_heads": 28,
+    "num_hidden_layers": 28,
+    "num_key_value_heads": 4,
+    "pad_token_id": null,
+    "rms_norm_eps": 1e-06,
+    "rope_parameters": {
+      "mrope_section": [
+        16,
+        24,
+        24
+      ],
+      "rope_theta": 1000000.0,
+      "rope_type": "default",
+      "type": "default"
+    },
+    "sliding_window": null,
+    "use_cache": true,
+    "use_sliding_window": false,
+    "vision_token_id": 151654,
+    "vocab_size": 152064
+  },
+  "tie_word_embeddings": false,
+  "transformers_version": "5.3.0",
+  "use_sliding_window": false,
+  "video_token_id": 151656,
+  "vision_config": {
+    "depth": 32,
+    "dtype": "bfloat16",
+    "fullatt_block_indexes": [
+      7,
+      15,
+      23,
+      31
+    ],
+    "hidden_act": "silu",
+    "hidden_size": 1280,
+    "in_channels": 3,
+    "in_chans": 3,
+    "initializer_range": 0.02,
+    "intermediate_size": 3420,
+    "model_type": "qwen2_5_vl",
+    "num_heads": 16,
+    "out_hidden_size": 3584,
+    "patch_size": 14,
+    "spatial_merge_size": 2,
+    "spatial_patch_size": 14,
+    "temporal_patch_size": 2,
+    "tokens_per_second": 2,
+    "window_size": 112
+  },
+  "vision_end_token_id": 151653,
+  "vision_start_token_id": 151652,
+  "vision_token_id": 151654,
+  "vocab_size": 152064
+}

text_encoder/generation_config.json

@@ -0,0 +1,14 @@
+{
+  "bos_token_id": 151643,
+  "do_sample": true,
+  "eos_token_id": [
+    151645,
+    151643
+  ],
+  "pad_token_id": 151643,
+  "repetition_penalty": 1.05,
+  "temperature": 0.1,
+  "top_k": 1,
+  "top_p": 0.001,
+  "transformers_version": "5.3.0"
+}

text_encoder/model.safetensors

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

tokenizer/added_tokens.json

@@ -0,0 +1,24 @@
+{
+  "</tool_call>": 151658,
+  "<tool_call>": 151657,
+  "<|box_end|>": 151649,
+  "<|box_start|>": 151648,
+  "<|endoftext|>": 151643,
+  "<|file_sep|>": 151664,
+  "<|fim_middle|>": 151660,
+  "<|fim_pad|>": 151662,
+  "<|fim_prefix|>": 151659,
+  "<|fim_suffix|>": 151661,
+  "<|im_end|>": 151645,
+  "<|im_start|>": 151644,
+  "<|image_pad|>": 151655,
+  "<|object_ref_end|>": 151647,
+  "<|object_ref_start|>": 151646,
+  "<|quad_end|>": 151651,
+  "<|quad_start|>": 151650,
+  "<|repo_name|>": 151663,
+  "<|video_pad|>": 151656,
+  "<|vision_end|>": 151653,
+  "<|vision_pad|>": 151654,
+  "<|vision_start|>": 151652
+}

tokenizer/chat_template.jinja

@@ -0,0 +1,54 @@
+{%- if tools %}
+    {{- '<|im_start|>system\n' }}
+    {%- if messages[0]['role'] == 'system' %}
+        {{- messages[0]['content'] }}
+    {%- else %}
+        {{- 'You are a helpful assistant.' }}
+    {%- endif %}
+    {{- "\n\n# Tools\n\nYou may call one or more functions to assist with the user query.\n\nYou are provided with function signatures within <tools></tools> XML tags:\n<tools>" }}
+    {%- for tool in tools %}
+        {{- "\n" }}
+        {{- tool | tojson }}
+    {%- endfor %}
+    {{- "\n</tools>\n\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\n<tool_call>\n{\"name\": <function-name>, \"arguments\": <args-json-object>}\n</tool_call><|im_end|>\n" }}
+{%- else %}
+    {%- if messages[0]['role'] == 'system' %}
+        {{- '<|im_start|>system\n' + messages[0]['content'] + '<|im_end|>\n' }}
+    {%- else %}
+        {{- '<|im_start|>system\nYou are a helpful assistant.<|im_end|>\n' }}
+    {%- endif %}
+{%- endif %}
+{%- for message in messages %}
+    {%- if (message.role == "user") or (message.role == "system" and not loop.first) or (message.role == "assistant" and not message.tool_calls) %}
+        {{- '<|im_start|>' + message.role + '\n' + message.content + '<|im_end|>' + '\n' }}
+    {%- elif message.role == "assistant" %}
+        {{- '<|im_start|>' + message.role }}
+        {%- if message.content %}
+            {{- '\n' + message.content }}
+        {%- endif %}
+        {%- for tool_call in message.tool_calls %}
+            {%- if tool_call.function is defined %}
+                {%- set tool_call = tool_call.function %}
+            {%- endif %}
+            {{- '\n<tool_call>\n{"name": "' }}
+            {{- tool_call.name }}
+            {{- '", "arguments": ' }}
+            {{- tool_call.arguments | tojson }}
+            {{- '}\n</tool_call>' }}
+        {%- endfor %}
+        {{- '<|im_end|>\n' }}
+    {%- elif message.role == "tool" %}
+        {%- if (loop.index0 == 0) or (messages[loop.index0 - 1].role != "tool") %}
+            {{- '<|im_start|>user' }}
+        {%- endif %}
+        {{- '\n<tool_response>\n' }}
+        {{- message.content }}
+        {{- '\n</tool_response>' }}
+        {%- if loop.last or (messages[loop.index0 + 1].role != "tool") %}
+            {{- '<|im_end|>\n' }}
+        {%- endif %}
+    {%- endif %}
+{%- endfor %}
+{%- if add_generation_prompt %}
+    {{- '<|im_start|>assistant\n' }}
+{%- endif %}

tokenizer/special_tokens_map.json

@@ -0,0 +1,31 @@
+{
+  "additional_special_tokens": [
+    "<|im_start|>",
+    "<|im_end|>",
+    "<|object_ref_start|>",
+    "<|object_ref_end|>",
+    "<|box_start|>",
+    "<|box_end|>",
+    "<|quad_start|>",
+    "<|quad_end|>",
+    "<|vision_start|>",
+    "<|vision_end|>",
+    "<|vision_pad|>",
+    "<|image_pad|>",
+    "<|video_pad|>"
+  ],
+  "eos_token": {
+    "content": "<|im_end|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

tokenizer/tokenizer_config.json

@@ -0,0 +1,207 @@
+{
+  "add_bos_token": false,
+  "add_prefix_space": false,
+  "added_tokens_decoder": {
+    "151643": {
+      "content": "<|endoftext|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151644": {
+      "content": "<|im_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151645": {
+      "content": "<|im_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151646": {
+      "content": "<|object_ref_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151647": {
+      "content": "<|object_ref_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151648": {
+      "content": "<|box_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151649": {
+      "content": "<|box_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151650": {
+      "content": "<|quad_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151651": {
+      "content": "<|quad_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151652": {
+      "content": "<|vision_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151653": {
+      "content": "<|vision_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151654": {
+      "content": "<|vision_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151655": {
+      "content": "<|image_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151656": {
+      "content": "<|video_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151657": {
+      "content": "<tool_call>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151658": {
+      "content": "</tool_call>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151659": {
+      "content": "<|fim_prefix|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151660": {
+      "content": "<|fim_middle|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151661": {
+      "content": "<|fim_suffix|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151662": {
+      "content": "<|fim_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151663": {
+      "content": "<|repo_name|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151664": {
+      "content": "<|file_sep|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    }
+  },
+  "additional_special_tokens": [
+    "<|im_start|>",
+    "<|im_end|>",
+    "<|object_ref_start|>",
+    "<|object_ref_end|>",
+    "<|box_start|>",
+    "<|box_end|>",
+    "<|quad_start|>",
+    "<|quad_end|>",
+    "<|vision_start|>",
+    "<|vision_end|>",
+    "<|vision_pad|>",
+    "<|image_pad|>",
+    "<|video_pad|>"
+  ],
+  "bos_token": null,
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "<|im_end|>",
+  "errors": "replace",
+  "extra_special_tokens": {},
+  "model_max_length": 131072,
+  "pad_token": "<|endoftext|>",
+  "split_special_tokens": false,
+  "tokenizer_class": "Qwen2Tokenizer",
+  "unk_token": null
+}

transformer/config.json

@@ -0,0 +1,1143 @@
+{
+  "_class_name": "QwenImageTransformer2DModel",
+  "_diffusers_version": "0.39.0.dev0",
+  "_name_or_path": "/home/ozzy/Documents/models/Qwen-Image-2512/transformer",
+  "attention_head_dim": 128,
+  "axes_dims_rope": [
+    16,
+    56,
+    56
+  ],
+  "guidance_embeds": false,
+  "in_channels": 64,
+  "joint_attention_dim": 3584,
+  "num_attention_heads": 24,
+  "num_layers": 60,
+  "out_channels": 16,
+  "patch_size": 2,
+  "quantization_config": {
+    "add_skip_keys": true,
+    "dequantize_fp32": false,
+    "dynamic_loss_threshold": 0.01,
+    "group_size": 0,
+    "hadamard_group_size": 256,
+    "is_integer": true,
+    "is_training": false,
+    "modules_dtype_dict": {
+      "uint4": [
+        "transformer_blocks.0.attn.add_k_proj.weight",
+        "transformer_blocks.0.attn.add_q_proj.weight",
+        "transformer_blocks.0.attn.add_v_proj.weight",
+        "transformer_blocks.0.attn.to_add_out.weight",
+        "transformer_blocks.0.attn.to_k.weight",
+        "transformer_blocks.0.attn.to_out.0.weight",
+        "transformer_blocks.0.attn.to_q.weight",
+        "transformer_blocks.0.attn.to_v.weight",
+        "transformer_blocks.0.img_mlp.net.0.proj.weight",
+        "transformer_blocks.0.img_mlp.net.2.weight",
+        "transformer_blocks.0.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.0.txt_mlp.net.2.weight",
+        "transformer_blocks.0.txt_mod.1.weight",
+        "transformer_blocks.1.attn.add_k_proj.weight",
+        "transformer_blocks.1.attn.add_q_proj.weight",
+        "transformer_blocks.1.attn.add_v_proj.weight",
+        "transformer_blocks.1.attn.to_add_out.weight",
+        "transformer_blocks.1.attn.to_k.weight",
+        "transformer_blocks.1.attn.to_out.0.weight",
+        "transformer_blocks.1.attn.to_q.weight",
+        "transformer_blocks.1.attn.to_v.weight",
+        "transformer_blocks.1.img_mlp.net.0.proj.weight",
+        "transformer_blocks.1.img_mlp.net.2.weight",
+        "transformer_blocks.1.img_mod.1.weight",
+        "transformer_blocks.1.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.1.txt_mlp.net.2.weight",
+        "transformer_blocks.1.txt_mod.1.weight",
+        "transformer_blocks.2.attn.add_k_proj.weight",
+        "transformer_blocks.2.attn.add_q_proj.weight",
+        "transformer_blocks.2.attn.add_v_proj.weight",
+        "transformer_blocks.2.attn.to_add_out.weight",
+        "transformer_blocks.2.attn.to_k.weight",
+        "transformer_blocks.2.attn.to_out.0.weight",
+        "transformer_blocks.2.attn.to_q.weight",
+        "transformer_blocks.2.attn.to_v.weight",
+        "transformer_blocks.2.img_mlp.net.0.proj.weight",
+        "transformer_blocks.2.img_mlp.net.2.weight",
+        "transformer_blocks.2.img_mod.1.weight",
+        "transformer_blocks.2.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.2.txt_mlp.net.2.weight",
+        "transformer_blocks.2.txt_mod.1.weight",
+        "transformer_blocks.3.attn.add_k_proj.weight",
+        "transformer_blocks.3.attn.add_q_proj.weight",
+        "transformer_blocks.3.attn.add_v_proj.weight",
+        "transformer_blocks.3.attn.to_add_out.weight",
+        "transformer_blocks.3.attn.to_k.weight",
+        "transformer_blocks.3.attn.to_out.0.weight",
+        "transformer_blocks.3.attn.to_q.weight",
+        "transformer_blocks.3.attn.to_v.weight",
+        "transformer_blocks.3.img_mlp.net.0.proj.weight",
+        "transformer_blocks.3.img_mlp.net.2.weight",
+        "transformer_blocks.3.img_mod.1.weight",
+        "transformer_blocks.3.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.3.txt_mlp.net.2.weight",
+        "transformer_blocks.3.txt_mod.1.weight",
+        "transformer_blocks.4.attn.add_k_proj.weight",
+        "transformer_blocks.4.attn.add_q_proj.weight",
+        "transformer_blocks.4.attn.add_v_proj.weight",
+        "transformer_blocks.4.attn.to_add_out.weight",
+        "transformer_blocks.4.attn.to_k.weight",
+        "transformer_blocks.4.attn.to_out.0.weight",
+        "transformer_blocks.4.attn.to_q.weight",
+        "transformer_blocks.4.attn.to_v.weight",
+        "transformer_blocks.4.img_mlp.net.0.proj.weight",
+        "transformer_blocks.4.img_mlp.net.2.weight",
+        "transformer_blocks.4.img_mod.1.weight",
+        "transformer_blocks.4.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.4.txt_mlp.net.2.weight",
+        "transformer_blocks.4.txt_mod.1.weight",
+        "transformer_blocks.5.attn.add_k_proj.weight",
+        "transformer_blocks.5.attn.add_q_proj.weight",
+        "transformer_blocks.5.attn.add_v_proj.weight",
+        "transformer_blocks.5.attn.to_add_out.weight",
+        "transformer_blocks.5.attn.to_k.weight",
+        "transformer_blocks.5.attn.to_out.0.weight",
+        "transformer_blocks.5.attn.to_q.weight",
+        "transformer_blocks.5.attn.to_v.weight",
+        "transformer_blocks.5.img_mlp.net.0.proj.weight",
+        "transformer_blocks.5.img_mlp.net.2.weight",
+        "transformer_blocks.5.img_mod.1.weight",
+        "transformer_blocks.5.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.5.txt_mlp.net.2.weight",
+        "transformer_blocks.5.txt_mod.1.weight",
+        "transformer_blocks.6.attn.add_k_proj.weight",
+        "transformer_blocks.6.attn.add_q_proj.weight",
+        "transformer_blocks.6.attn.add_v_proj.weight",
+        "transformer_blocks.6.attn.to_add_out.weight",
+        "transformer_blocks.6.attn.to_k.weight",
+        "transformer_blocks.6.attn.to_out.0.weight",
+        "transformer_blocks.6.attn.to_q.weight",
+        "transformer_blocks.6.attn.to_v.weight",
+        "transformer_blocks.6.img_mlp.net.0.proj.weight",
+        "transformer_blocks.6.img_mlp.net.2.weight",
+        "transformer_blocks.6.img_mod.1.weight",
+        "transformer_blocks.6.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.6.txt_mlp.net.2.weight",
+        "transformer_blocks.6.txt_mod.1.weight",
+        "transformer_blocks.7.attn.add_k_proj.weight",
+        "transformer_blocks.7.attn.to_k.weight",
+        "transformer_blocks.7.attn.to_q.weight",
+        "transformer_blocks.7.attn.to_v.weight",
+        "transformer_blocks.7.img_mod.1.weight",
+        "transformer_blocks.10.attn.add_k_proj.weight",
+        "transformer_blocks.10.attn.add_q_proj.weight",
+        "transformer_blocks.10.attn.add_v_proj.weight",
+        "transformer_blocks.10.attn.to_add_out.weight",
+        "transformer_blocks.10.attn.to_k.weight",
+        "transformer_blocks.10.attn.to_out.0.weight",
+        "transformer_blocks.10.attn.to_q.weight",
+        "transformer_blocks.10.attn.to_v.weight",
+        "transformer_blocks.10.img_mlp.net.0.proj.weight",
+        "transformer_blocks.10.img_mlp.net.2.weight",
+        "transformer_blocks.10.img_mod.1.weight",
+        "transformer_blocks.10.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.10.txt_mlp.net.2.weight",
+        "transformer_blocks.10.txt_mod.1.weight",
+        "transformer_blocks.11.attn.add_k_proj.weight",
+        "transformer_blocks.11.attn.add_q_proj.weight",
+        "transformer_blocks.11.attn.add_v_proj.weight",
+        "transformer_blocks.11.attn.to_add_out.weight",
+        "transformer_blocks.11.attn.to_k.weight",
+        "transformer_blocks.11.attn.to_out.0.weight",
+        "transformer_blocks.11.attn.to_q.weight",
+        "transformer_blocks.11.attn.to_v.weight",
+        "transformer_blocks.11.img_mlp.net.0.proj.weight",
+        "transformer_blocks.11.img_mlp.net.2.weight",
+        "transformer_blocks.11.img_mod.1.weight",
+        "transformer_blocks.11.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.11.txt_mlp.net.2.weight",
+        "transformer_blocks.11.txt_mod.1.weight",
+        "transformer_blocks.12.attn.add_k_proj.weight",
+        "transformer_blocks.12.attn.add_q_proj.weight",
+        "transformer_blocks.12.attn.add_v_proj.weight",
+        "transformer_blocks.12.attn.to_add_out.weight",
+        "transformer_blocks.12.attn.to_k.weight",
+        "transformer_blocks.12.attn.to_out.0.weight",
+        "transformer_blocks.12.attn.to_q.weight",
+        "transformer_blocks.12.attn.to_v.weight",
+        "transformer_blocks.12.img_mlp.net.0.proj.weight",
+        "transformer_blocks.12.img_mlp.net.2.weight",
+        "transformer_blocks.12.img_mod.1.weight",
+        "transformer_blocks.12.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.12.txt_mlp.net.2.weight",
+        "transformer_blocks.12.txt_mod.1.weight",
+        "transformer_blocks.13.attn.add_k_proj.weight",
+        "transformer_blocks.13.attn.add_q_proj.weight",
+        "transformer_blocks.13.attn.add_v_proj.weight",
+        "transformer_blocks.13.attn.to_add_out.weight",
+        "transformer_blocks.13.attn.to_k.weight",
+        "transformer_blocks.13.attn.to_out.0.weight",
+        "transformer_blocks.13.attn.to_q.weight",
+        "transformer_blocks.13.attn.to_v.weight",
+        "transformer_blocks.13.img_mlp.net.0.proj.weight",
+        "transformer_blocks.13.img_mlp.net.2.weight",
+        "transformer_blocks.13.img_mod.1.weight",
+        "transformer_blocks.13.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.13.txt_mlp.net.2.weight",
+        "transformer_blocks.13.txt_mod.1.weight",
+        "transformer_blocks.14.attn.add_k_proj.weight",
+        "transformer_blocks.14.attn.add_q_proj.weight",
+        "transformer_blocks.14.attn.add_v_proj.weight",
+        "transformer_blocks.14.attn.to_add_out.weight",
+        "transformer_blocks.14.attn.to_k.weight",
+        "transformer_blocks.14.attn.to_out.0.weight",
+        "transformer_blocks.14.attn.to_q.weight",
+        "transformer_blocks.14.attn.to_v.weight",
+        "transformer_blocks.14.img_mlp.net.0.proj.weight",
+        "transformer_blocks.14.img_mlp.net.2.weight",
+        "transformer_blocks.14.img_mod.1.weight",
+        "transformer_blocks.7.attn.add_q_proj.weight",
+        "transformer_blocks.7.attn.add_v_proj.weight",
+        "transformer_blocks.7.attn.to_add_out.weight",
+        "transformer_blocks.7.attn.to_out.0.weight",
+        "transformer_blocks.7.img_mlp.net.0.proj.weight",
+        "transformer_blocks.7.img_mlp.net.2.weight",
+        "transformer_blocks.7.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.7.txt_mlp.net.2.weight",
+        "transformer_blocks.7.txt_mod.1.weight",
+        "transformer_blocks.8.attn.add_k_proj.weight",
+        "transformer_blocks.8.attn.add_q_proj.weight",
+        "transformer_blocks.8.attn.add_v_proj.weight",
+        "transformer_blocks.8.attn.to_add_out.weight",
+        "transformer_blocks.8.attn.to_k.weight",
+        "transformer_blocks.8.attn.to_out.0.weight",
+        "transformer_blocks.8.attn.to_q.weight",
+        "transformer_blocks.8.attn.to_v.weight",
+        "transformer_blocks.8.img_mlp.net.0.proj.weight",
+        "transformer_blocks.8.img_mlp.net.2.weight",
+        "transformer_blocks.8.img_mod.1.weight",
+        "transformer_blocks.8.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.8.txt_mlp.net.2.weight",
+        "transformer_blocks.8.txt_mod.1.weight",
+        "transformer_blocks.9.attn.add_k_proj.weight",
+        "transformer_blocks.9.attn.add_q_proj.weight",
+        "transformer_blocks.9.attn.add_v_proj.weight",
+        "transformer_blocks.9.attn.to_add_out.weight",
+        "transformer_blocks.9.attn.to_k.weight",
+        "transformer_blocks.9.attn.to_out.0.weight",
+        "transformer_blocks.9.attn.to_q.weight",
+        "transformer_blocks.9.attn.to_v.weight",
+        "transformer_blocks.9.img_mlp.net.0.proj.weight",
+        "transformer_blocks.9.img_mlp.net.2.weight",
+        "transformer_blocks.9.img_mod.1.weight",
+        "transformer_blocks.9.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.9.txt_mlp.net.2.weight",
+        "transformer_blocks.9.txt_mod.1.weight",
+        "transformer_blocks.14.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.14.txt_mlp.net.2.weight",
+        "transformer_blocks.14.txt_mod.1.weight",
+        "transformer_blocks.15.attn.add_k_proj.weight",
+        "transformer_blocks.15.attn.add_q_proj.weight",
+        "transformer_blocks.15.attn.add_v_proj.weight",
+        "transformer_blocks.15.attn.to_add_out.weight",
+        "transformer_blocks.15.attn.to_k.weight",
+        "transformer_blocks.15.attn.to_out.0.weight",
+        "transformer_blocks.15.attn.to_q.weight",
+        "transformer_blocks.15.attn.to_v.weight",
+        "transformer_blocks.15.img_mlp.net.0.proj.weight",
+        "transformer_blocks.15.img_mlp.net.2.weight",
+        "transformer_blocks.15.img_mod.1.weight",
+        "transformer_blocks.15.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.15.txt_mlp.net.2.weight",
+        "transformer_blocks.15.txt_mod.1.weight",
+        "transformer_blocks.16.attn.add_k_proj.weight",
+        "transformer_blocks.16.attn.add_q_proj.weight",
+        "transformer_blocks.16.attn.add_v_proj.weight",
+        "transformer_blocks.16.attn.to_add_out.weight",
+        "transformer_blocks.16.attn.to_k.weight",
+        "transformer_blocks.16.attn.to_out.0.weight",
+        "transformer_blocks.16.attn.to_q.weight",
+        "transformer_blocks.16.attn.to_v.weight",
+        "transformer_blocks.16.img_mlp.net.0.proj.weight",
+        "transformer_blocks.16.img_mlp.net.2.weight",
+        "transformer_blocks.16.img_mod.1.weight",
+        "transformer_blocks.16.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.16.txt_mlp.net.2.weight",
+        "transformer_blocks.16.txt_mod.1.weight",
+        "transformer_blocks.17.attn.add_k_proj.weight",
+        "transformer_blocks.17.attn.add_q_proj.weight",
+        "transformer_blocks.17.attn.add_v_proj.weight",
+        "transformer_blocks.17.attn.to_add_out.weight",
+        "transformer_blocks.17.attn.to_k.weight",
+        "transformer_blocks.17.attn.to_out.0.weight",
+        "transformer_blocks.17.attn.to_q.weight",
+        "transformer_blocks.17.attn.to_v.weight",
+        "transformer_blocks.17.img_mlp.net.0.proj.weight",
+        "transformer_blocks.17.img_mlp.net.2.weight",
+        "transformer_blocks.17.img_mod.1.weight",
+        "transformer_blocks.17.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.17.txt_mlp.net.2.weight",
+        "transformer_blocks.17.txt_mod.1.weight",
+        "transformer_blocks.18.attn.add_k_proj.weight",
+        "transformer_blocks.18.attn.add_q_proj.weight",
+        "transformer_blocks.18.attn.add_v_proj.weight",
+        "transformer_blocks.18.attn.to_add_out.weight",
+        "transformer_blocks.18.attn.to_k.weight",
+        "transformer_blocks.18.attn.to_out.0.weight",
+        "transformer_blocks.18.attn.to_q.weight",
+        "transformer_blocks.18.attn.to_v.weight",
+        "transformer_blocks.18.img_mlp.net.0.proj.weight",
+        "transformer_blocks.18.img_mlp.net.2.weight",
+        "transformer_blocks.18.img_mod.1.weight",
+        "transformer_blocks.18.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.18.txt_mlp.net.2.weight",
+        "transformer_blocks.18.txt_mod.1.weight",
+        "transformer_blocks.19.attn.add_k_proj.weight",
+        "transformer_blocks.19.attn.add_q_proj.weight",
+        "transformer_blocks.19.attn.add_v_proj.weight",
+        "transformer_blocks.19.attn.to_add_out.weight",
+        "transformer_blocks.19.attn.to_k.weight",
+        "transformer_blocks.19.attn.to_out.0.weight",
+        "transformer_blocks.19.attn.to_q.weight",
+        "transformer_blocks.19.attn.to_v.weight",
+        "transformer_blocks.19.img_mlp.net.0.proj.weight",
+        "transformer_blocks.19.img_mlp.net.2.weight",
+        "transformer_blocks.19.img_mod.1.weight",
+        "transformer_blocks.19.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.19.txt_mlp.net.2.weight",
+        "transformer_blocks.19.txt_mod.1.weight",
+        "transformer_blocks.20.attn.add_k_proj.weight",
+        "transformer_blocks.20.attn.add_q_proj.weight",
+        "transformer_blocks.20.attn.add_v_proj.weight",
+        "transformer_blocks.20.attn.to_add_out.weight",
+        "transformer_blocks.20.attn.to_k.weight",
+        "transformer_blocks.20.attn.to_out.0.weight",
+        "transformer_blocks.20.attn.to_q.weight",
+        "transformer_blocks.20.attn.to_v.weight",
+        "transformer_blocks.20.img_mlp.net.0.proj.weight",
+        "transformer_blocks.20.img_mlp.net.2.weight",
+        "transformer_blocks.20.img_mod.1.weight",
+        "transformer_blocks.20.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.20.txt_mlp.net.2.weight",
+        "transformer_blocks.20.txt_mod.1.weight",
+        "transformer_blocks.21.attn.add_k_proj.weight",
+        "transformer_blocks.21.attn.add_q_proj.weight",
+        "transformer_blocks.21.attn.add_v_proj.weight",
+        "transformer_blocks.21.attn.to_add_out.weight",
+        "transformer_blocks.21.attn.to_k.weight",
+        "transformer_blocks.21.attn.to_out.0.weight",
+        "transformer_blocks.21.attn.to_q.weight",
+        "transformer_blocks.21.attn.to_v.weight",
+        "transformer_blocks.21.img_mlp.net.0.proj.weight",
+        "transformer_blocks.21.img_mlp.net.2.weight",
+        "transformer_blocks.21.img_mod.1.weight",
+        "transformer_blocks.21.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.21.txt_mod.1.weight",
+        "transformer_blocks.21.txt_mlp.net.2.weight",
+        "transformer_blocks.22.attn.add_k_proj.weight",
+        "transformer_blocks.22.attn.add_q_proj.weight",
+        "transformer_blocks.22.attn.add_v_proj.weight",
+        "transformer_blocks.22.attn.to_add_out.weight",
+        "transformer_blocks.22.attn.to_k.weight",
+        "transformer_blocks.22.attn.to_out.0.weight",
+        "transformer_blocks.22.attn.to_q.weight",
+        "transformer_blocks.22.attn.to_v.weight",
+        "transformer_blocks.22.img_mlp.net.0.proj.weight",
+        "transformer_blocks.22.img_mlp.net.2.weight",
+        "transformer_blocks.22.img_mod.1.weight",
+        "transformer_blocks.22.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.22.txt_mlp.net.2.weight",
+        "transformer_blocks.22.txt_mod.1.weight",
+        "transformer_blocks.23.attn.add_k_proj.weight",
+        "transformer_blocks.23.attn.add_q_proj.weight",
+        "transformer_blocks.23.attn.add_v_proj.weight",
+        "transformer_blocks.23.attn.to_add_out.weight",
+        "transformer_blocks.23.attn.to_k.weight",
+        "transformer_blocks.23.attn.to_out.0.weight",
+        "transformer_blocks.23.attn.to_q.weight",
+        "transformer_blocks.23.attn.to_v.weight",
+        "transformer_blocks.23.img_mlp.net.0.proj.weight",
+        "transformer_blocks.23.img_mlp.net.2.weight",
+        "transformer_blocks.23.img_mod.1.weight",
+        "transformer_blocks.23.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.23.txt_mlp.net.2.weight",
+        "transformer_blocks.23.txt_mod.1.weight",
+        "transformer_blocks.24.attn.add_k_proj.weight",
+        "transformer_blocks.24.attn.add_q_proj.weight",
+        "transformer_blocks.24.attn.add_v_proj.weight",
+        "transformer_blocks.24.attn.to_add_out.weight",
+        "transformer_blocks.24.attn.to_k.weight",
+        "transformer_blocks.24.attn.to_out.0.weight",
+        "transformer_blocks.24.attn.to_q.weight",
+        "transformer_blocks.24.attn.to_v.weight",
+        "transformer_blocks.24.img_mlp.net.0.proj.weight",
+        "transformer_blocks.24.img_mlp.net.2.weight",
+        "transformer_blocks.24.img_mod.1.weight",
+        "transformer_blocks.24.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.24.txt_mlp.net.2.weight",
+        "transformer_blocks.24.txt_mod.1.weight",
+        "transformer_blocks.25.attn.add_k_proj.weight",
+        "transformer_blocks.25.attn.add_q_proj.weight",
+        "transformer_blocks.25.attn.add_v_proj.weight",
+        "transformer_blocks.25.attn.to_add_out.weight",
+        "transformer_blocks.25.attn.to_k.weight",
+        "transformer_blocks.25.attn.to_out.0.weight",
+        "transformer_blocks.25.attn.to_q.weight",
+        "transformer_blocks.25.attn.to_v.weight",
+        "transformer_blocks.25.img_mlp.net.0.proj.weight",
+        "transformer_blocks.25.img_mlp.net.2.weight",
+        "transformer_blocks.25.img_mod.1.weight",
+        "transformer_blocks.25.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.25.txt_mlp.net.2.weight",
+        "transformer_blocks.25.txt_mod.1.weight",
+        "transformer_blocks.26.attn.add_k_proj.weight",
+        "transformer_blocks.26.attn.add_q_proj.weight",
+        "transformer_blocks.26.attn.add_v_proj.weight",
+        "transformer_blocks.26.attn.to_add_out.weight",
+        "transformer_blocks.26.attn.to_k.weight",
+        "transformer_blocks.26.attn.to_out.0.weight",
+        "transformer_blocks.26.attn.to_q.weight",
+        "transformer_blocks.26.attn.to_v.weight",
+        "transformer_blocks.26.img_mlp.net.0.proj.weight",
+        "transformer_blocks.26.img_mlp.net.2.weight",
+        "transformer_blocks.26.img_mod.1.weight",
+        "transformer_blocks.26.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.26.txt_mlp.net.2.weight",
+        "transformer_blocks.26.txt_mod.1.weight",
+        "transformer_blocks.27.attn.add_k_proj.weight",
+        "transformer_blocks.27.attn.add_q_proj.weight",
+        "transformer_blocks.27.attn.add_v_proj.weight",
+        "transformer_blocks.27.attn.to_add_out.weight",
+        "transformer_blocks.27.attn.to_k.weight",
+        "transformer_blocks.27.attn.to_out.0.weight",
+        "transformer_blocks.27.attn.to_q.weight",
+        "transformer_blocks.27.attn.to_v.weight",
+        "transformer_blocks.27.img_mlp.net.0.proj.weight",
+        "transformer_blocks.27.img_mlp.net.2.weight",
+        "transformer_blocks.27.img_mod.1.weight",
+        "transformer_blocks.27.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.27.txt_mlp.net.2.weight",
+        "transformer_blocks.27.txt_mod.1.weight",
+        "transformer_blocks.28.attn.add_k_proj.weight",
+        "transformer_blocks.28.attn.add_q_proj.weight",
+        "transformer_blocks.28.attn.add_v_proj.weight",
+        "transformer_blocks.28.attn.to_add_out.weight",
+        "transformer_blocks.28.attn.to_k.weight",
+        "transformer_blocks.28.attn.to_out.0.weight",
+        "transformer_blocks.28.attn.to_q.weight",
+        "transformer_blocks.28.attn.to_v.weight",
+        "transformer_blocks.28.img_mlp.net.0.proj.weight",
+        "transformer_blocks.28.img_mlp.net.2.weight",
+        "transformer_blocks.28.img_mod.1.weight",
+        "transformer_blocks.28.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.28.txt_mlp.net.2.weight",
+        "transformer_blocks.28.txt_mod.1.weight",
+        "transformer_blocks.29.attn.to_k.weight",
+        "transformer_blocks.29.attn.to_q.weight",
+        "transformer_blocks.29.img_mod.1.weight",
+        "transformer_blocks.29.attn.add_k_proj.weight",
+        "transformer_blocks.29.attn.add_q_proj.weight",
+        "transformer_blocks.29.attn.add_v_proj.weight",
+        "transformer_blocks.29.attn.to_add_out.weight",
+        "transformer_blocks.29.attn.to_out.0.weight",
+        "transformer_blocks.29.attn.to_v.weight",
+        "transformer_blocks.29.img_mlp.net.0.proj.weight",
+        "transformer_blocks.29.img_mlp.net.2.weight",
+        "transformer_blocks.29.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.29.txt_mlp.net.2.weight",
+        "transformer_blocks.29.txt_mod.1.weight",
+        "transformer_blocks.30.attn.add_k_proj.weight",
+        "transformer_blocks.30.attn.add_q_proj.weight",
+        "transformer_blocks.30.attn.add_v_proj.weight",
+        "transformer_blocks.30.attn.to_add_out.weight",
+        "transformer_blocks.30.attn.to_k.weight",
+        "transformer_blocks.30.attn.to_out.0.weight",
+        "transformer_blocks.30.attn.to_q.weight",
+        "transformer_blocks.30.attn.to_v.weight",
+        "transformer_blocks.30.img_mlp.net.0.proj.weight",
+        "transformer_blocks.30.img_mlp.net.2.weight",
+        "transformer_blocks.30.img_mod.1.weight",
+        "transformer_blocks.30.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.30.txt_mlp.net.2.weight",
+        "transformer_blocks.30.txt_mod.1.weight",
+        "transformer_blocks.31.attn.add_k_proj.weight",
+        "transformer_blocks.31.attn.add_q_proj.weight",
+        "transformer_blocks.31.attn.add_v_proj.weight",
+        "transformer_blocks.31.attn.to_add_out.weight",
+        "transformer_blocks.31.attn.to_k.weight",
+        "transformer_blocks.31.attn.to_out.0.weight",
+        "transformer_blocks.31.attn.to_q.weight",
+        "transformer_blocks.31.attn.to_v.weight",
+        "transformer_blocks.31.img_mlp.net.0.proj.weight",
+        "transformer_blocks.31.img_mlp.net.2.weight",
+        "transformer_blocks.31.img_mod.1.weight",
+        "transformer_blocks.31.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.31.txt_mlp.net.2.weight",
+        "transformer_blocks.31.txt_mod.1.weight",
+        "transformer_blocks.32.attn.add_k_proj.weight",
+        "transformer_blocks.32.attn.add_q_proj.weight",
+        "transformer_blocks.32.attn.add_v_proj.weight",
+        "transformer_blocks.32.attn.to_add_out.weight",
+        "transformer_blocks.32.attn.to_k.weight",
+        "transformer_blocks.32.attn.to_out.0.weight",
+        "transformer_blocks.32.attn.to_q.weight",
+        "transformer_blocks.32.attn.to_v.weight",
+        "transformer_blocks.32.img_mlp.net.0.proj.weight",
+        "transformer_blocks.32.img_mlp.net.2.weight",
+        "transformer_blocks.32.img_mod.1.weight",
+        "transformer_blocks.32.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.32.txt_mlp.net.2.weight",
+        "transformer_blocks.32.txt_mod.1.weight",
+        "transformer_blocks.33.attn.add_k_proj.weight",
+        "transformer_blocks.33.attn.add_q_proj.weight",
+        "transformer_blocks.33.attn.add_v_proj.weight",
+        "transformer_blocks.33.attn.to_add_out.weight",
+        "transformer_blocks.33.attn.to_k.weight",
+        "transformer_blocks.33.attn.to_out.0.weight",
+        "transformer_blocks.33.attn.to_q.weight",
+        "transformer_blocks.33.attn.to_v.weight",
+        "transformer_blocks.33.img_mlp.net.0.proj.weight",
+        "transformer_blocks.33.img_mlp.net.2.weight",
+        "transformer_blocks.33.img_mod.1.weight",
+        "transformer_blocks.33.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.33.txt_mlp.net.2.weight",
+        "transformer_blocks.33.txt_mod.1.weight",
+        "transformer_blocks.34.attn.add_k_proj.weight",
+        "transformer_blocks.34.attn.add_q_proj.weight",
+        "transformer_blocks.34.attn.add_v_proj.weight",
+        "transformer_blocks.34.attn.to_add_out.weight",
+        "transformer_blocks.34.attn.to_k.weight",
+        "transformer_blocks.34.attn.to_out.0.weight",
+        "transformer_blocks.34.attn.to_q.weight",
+        "transformer_blocks.34.attn.to_v.weight",
+        "transformer_blocks.34.img_mlp.net.0.proj.weight",
+        "transformer_blocks.34.img_mlp.net.2.weight",
+        "transformer_blocks.34.img_mod.1.weight",
+        "transformer_blocks.34.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.34.txt_mlp.net.2.weight",
+        "transformer_blocks.34.txt_mod.1.weight",
+        "transformer_blocks.35.attn.add_k_proj.weight",
+        "transformer_blocks.35.attn.add_q_proj.weight",
+        "transformer_blocks.35.attn.add_v_proj.weight",
+        "transformer_blocks.35.attn.to_add_out.weight",
+        "transformer_blocks.35.attn.to_k.weight",
+        "transformer_blocks.35.attn.to_out.0.weight",
+        "transformer_blocks.35.attn.to_q.weight",
+        "transformer_blocks.35.attn.to_v.weight",
+        "transformer_blocks.35.img_mlp.net.0.proj.weight",
+        "transformer_blocks.35.img_mlp.net.2.weight",
+        "transformer_blocks.35.img_mod.1.weight",
+        "transformer_blocks.35.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.35.txt_mlp.net.2.weight",
+        "transformer_blocks.35.txt_mod.1.weight",
+        "transformer_blocks.36.attn.add_k_proj.weight",
+        "transformer_blocks.36.attn.add_q_proj.weight",
+        "transformer_blocks.36.attn.add_v_proj.weight",
+        "transformer_blocks.36.attn.to_add_out.weight",
+        "transformer_blocks.36.attn.to_k.weight",
+        "transformer_blocks.36.attn.to_out.0.weight",
+        "transformer_blocks.36.attn.to_q.weight",
+        "transformer_blocks.36.attn.to_v.weight",
+        "transformer_blocks.36.img_mlp.net.0.proj.weight",
+        "transformer_blocks.36.img_mod.1.weight",
+        "transformer_blocks.36.img_mlp.net.2.weight",
+        "transformer_blocks.36.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.36.txt_mlp.net.2.weight",
+        "transformer_blocks.36.txt_mod.1.weight",
+        "transformer_blocks.37.attn.add_k_proj.weight",
+        "transformer_blocks.37.attn.add_q_proj.weight",
+        "transformer_blocks.37.attn.add_v_proj.weight",
+        "transformer_blocks.37.attn.to_add_out.weight",
+        "transformer_blocks.37.attn.to_k.weight",
+        "transformer_blocks.37.attn.to_out.0.weight",
+        "transformer_blocks.37.attn.to_q.weight",
+        "transformer_blocks.37.attn.to_v.weight",
+        "transformer_blocks.37.img_mlp.net.0.proj.weight",
+        "transformer_blocks.37.img_mlp.net.2.weight",
+        "transformer_blocks.37.img_mod.1.weight",
+        "transformer_blocks.37.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.37.txt_mlp.net.2.weight",
+        "transformer_blocks.37.txt_mod.1.weight",
+        "transformer_blocks.38.attn.add_k_proj.weight",
+        "transformer_blocks.38.attn.add_q_proj.weight",
+        "transformer_blocks.38.attn.add_v_proj.weight",
+        "transformer_blocks.38.attn.to_add_out.weight",
+        "transformer_blocks.38.attn.to_k.weight",
+        "transformer_blocks.38.attn.to_out.0.weight",
+        "transformer_blocks.38.attn.to_q.weight",
+        "transformer_blocks.38.attn.to_v.weight",
+        "transformer_blocks.38.img_mlp.net.0.proj.weight",
+        "transformer_blocks.38.img_mlp.net.2.weight",
+        "transformer_blocks.38.img_mod.1.weight",
+        "transformer_blocks.38.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.38.txt_mlp.net.2.weight",
+        "transformer_blocks.38.txt_mod.1.weight",
+        "transformer_blocks.39.attn.add_k_proj.weight",
+        "transformer_blocks.39.attn.add_q_proj.weight",
+        "transformer_blocks.39.attn.add_v_proj.weight",
+        "transformer_blocks.39.attn.to_add_out.weight",
+        "transformer_blocks.39.attn.to_k.weight",
+        "transformer_blocks.39.attn.to_out.0.weight",
+        "transformer_blocks.39.attn.to_q.weight",
+        "transformer_blocks.39.attn.to_v.weight",
+        "transformer_blocks.39.img_mlp.net.0.proj.weight",
+        "transformer_blocks.39.img_mlp.net.2.weight",
+        "transformer_blocks.39.img_mod.1.weight",
+        "transformer_blocks.39.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.39.txt_mlp.net.2.weight",
+        "transformer_blocks.39.txt_mod.1.weight",
+        "transformer_blocks.40.attn.add_k_proj.weight",
+        "transformer_blocks.40.attn.add_q_proj.weight",
+        "transformer_blocks.40.attn.add_v_proj.weight",
+        "transformer_blocks.40.attn.to_add_out.weight",
+        "transformer_blocks.40.attn.to_k.weight",
+        "transformer_blocks.40.attn.to_out.0.weight",
+        "transformer_blocks.40.attn.to_q.weight",
+        "transformer_blocks.40.attn.to_v.weight",
+        "transformer_blocks.40.img_mlp.net.0.proj.weight",
+        "transformer_blocks.40.img_mlp.net.2.weight",
+        "transformer_blocks.40.img_mod.1.weight",
+        "transformer_blocks.40.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.40.txt_mlp.net.2.weight",
+        "transformer_blocks.40.txt_mod.1.weight",
+        "transformer_blocks.41.attn.add_k_proj.weight",
+        "transformer_blocks.41.attn.add_q_proj.weight",
+        "transformer_blocks.41.attn.add_v_proj.weight",
+        "transformer_blocks.41.attn.to_add_out.weight",
+        "transformer_blocks.41.attn.to_k.weight",
+        "transformer_blocks.41.attn.to_out.0.weight",
+        "transformer_blocks.41.attn.to_q.weight",
+        "transformer_blocks.41.attn.to_v.weight",
+        "transformer_blocks.41.img_mlp.net.0.proj.weight",
+        "transformer_blocks.41.img_mlp.net.2.weight",
+        "transformer_blocks.41.img_mod.1.weight",
+        "transformer_blocks.41.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.41.txt_mlp.net.2.weight",
+        "transformer_blocks.41.txt_mod.1.weight",
+        "transformer_blocks.42.attn.add_k_proj.weight",
+        "transformer_blocks.42.attn.add_q_proj.weight",
+        "transformer_blocks.42.attn.add_v_proj.weight",
+        "transformer_blocks.42.attn.to_add_out.weight",
+        "transformer_blocks.42.attn.to_k.weight",
+        "transformer_blocks.42.attn.to_out.0.weight",
+        "transformer_blocks.42.attn.to_q.weight",
+        "transformer_blocks.42.attn.to_v.weight",
+        "transformer_blocks.42.img_mlp.net.0.proj.weight",
+        "transformer_blocks.42.img_mlp.net.2.weight",
+        "transformer_blocks.42.img_mod.1.weight",
+        "transformer_blocks.42.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.42.txt_mlp.net.2.weight",
+        "transformer_blocks.42.txt_mod.1.weight",
+        "transformer_blocks.43.attn.add_k_proj.weight",
+        "transformer_blocks.43.attn.add_q_proj.weight",
+        "transformer_blocks.43.attn.add_v_proj.weight",
+        "transformer_blocks.43.attn.to_add_out.weight",
+        "transformer_blocks.43.attn.to_k.weight",
+        "transformer_blocks.43.attn.to_out.0.weight",
+        "transformer_blocks.43.attn.to_q.weight",
+        "transformer_blocks.43.attn.to_v.weight",
+        "transformer_blocks.43.img_mlp.net.0.proj.weight",
+        "transformer_blocks.43.img_mlp.net.2.weight",
+        "transformer_blocks.43.img_mod.1.weight",
+        "transformer_blocks.43.txt_mod.1.weight",
+        "transformer_blocks.43.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.43.txt_mlp.net.2.weight",
+        "transformer_blocks.44.attn.add_k_proj.weight",
+        "transformer_blocks.44.attn.add_q_proj.weight",
+        "transformer_blocks.44.attn.add_v_proj.weight",
+        "transformer_blocks.44.attn.to_add_out.weight",
+        "transformer_blocks.44.attn.to_k.weight",
+        "transformer_blocks.44.attn.to_out.0.weight",
+        "transformer_blocks.44.attn.to_q.weight",
+        "transformer_blocks.44.attn.to_v.weight",
+        "transformer_blocks.44.img_mlp.net.0.proj.weight",
+        "transformer_blocks.44.img_mlp.net.2.weight",
+        "transformer_blocks.44.img_mod.1.weight",
+        "transformer_blocks.44.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.44.txt_mlp.net.2.weight",
+        "transformer_blocks.44.txt_mod.1.weight",
+        "transformer_blocks.45.attn.add_k_proj.weight",
+        "transformer_blocks.45.attn.add_q_proj.weight",
+        "transformer_blocks.45.attn.add_v_proj.weight",
+        "transformer_blocks.45.attn.to_add_out.weight",
+        "transformer_blocks.45.attn.to_k.weight",
+        "transformer_blocks.45.attn.to_out.0.weight",
+        "transformer_blocks.45.attn.to_q.weight",
+        "transformer_blocks.45.attn.to_v.weight",
+        "transformer_blocks.45.img_mlp.net.0.proj.weight",
+        "transformer_blocks.45.img_mlp.net.2.weight",
+        "transformer_blocks.45.img_mod.1.weight",
+        "transformer_blocks.45.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.45.txt_mlp.net.2.weight",
+        "transformer_blocks.45.txt_mod.1.weight",
+        "transformer_blocks.46.attn.add_k_proj.weight",
+        "transformer_blocks.46.attn.add_q_proj.weight",
+        "transformer_blocks.46.attn.add_v_proj.weight",
+        "transformer_blocks.46.attn.to_add_out.weight",
+        "transformer_blocks.46.attn.to_k.weight",
+        "transformer_blocks.46.attn.to_out.0.weight",
+        "transformer_blocks.46.attn.to_q.weight",
+        "transformer_blocks.46.attn.to_v.weight",
+        "transformer_blocks.46.img_mlp.net.0.proj.weight",
+        "transformer_blocks.46.img_mlp.net.2.weight",
+        "transformer_blocks.46.img_mod.1.weight",
+        "transformer_blocks.46.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.46.txt_mlp.net.2.weight",
+        "transformer_blocks.46.txt_mod.1.weight",
+        "transformer_blocks.47.attn.add_k_proj.weight",
+        "transformer_blocks.47.attn.add_q_proj.weight",
+        "transformer_blocks.47.attn.add_v_proj.weight",
+        "transformer_blocks.47.attn.to_add_out.weight",
+        "transformer_blocks.47.attn.to_k.weight",
+        "transformer_blocks.47.attn.to_out.0.weight",
+        "transformer_blocks.47.attn.to_q.weight",
+        "transformer_blocks.47.attn.to_v.weight",
+        "transformer_blocks.47.img_mlp.net.0.proj.weight",
+        "transformer_blocks.47.img_mlp.net.2.weight",
+        "transformer_blocks.47.img_mod.1.weight",
+        "transformer_blocks.47.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.47.txt_mlp.net.2.weight",
+        "transformer_blocks.47.txt_mod.1.weight",
+        "transformer_blocks.48.attn.add_k_proj.weight",
+        "transformer_blocks.48.attn.add_q_proj.weight",
+        "transformer_blocks.48.attn.add_v_proj.weight",
+        "transformer_blocks.48.attn.to_add_out.weight",
+        "transformer_blocks.48.attn.to_k.weight",
+        "transformer_blocks.48.attn.to_out.0.weight",
+        "transformer_blocks.48.attn.to_q.weight",
+        "transformer_blocks.48.attn.to_v.weight",
+        "transformer_blocks.48.img_mlp.net.0.proj.weight",
+        "transformer_blocks.48.img_mlp.net.2.weight",
+        "transformer_blocks.48.img_mod.1.weight",
+        "transformer_blocks.48.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.48.txt_mlp.net.2.weight",
+        "transformer_blocks.48.txt_mod.1.weight",
+        "transformer_blocks.49.attn.add_k_proj.weight",
+        "transformer_blocks.49.attn.add_q_proj.weight",
+        "transformer_blocks.49.attn.add_v_proj.weight",
+        "transformer_blocks.49.attn.to_add_out.weight",
+        "transformer_blocks.49.attn.to_k.weight",
+        "transformer_blocks.49.attn.to_out.0.weight",
+        "transformer_blocks.49.attn.to_q.weight",
+        "transformer_blocks.49.attn.to_v.weight",
+        "transformer_blocks.49.img_mlp.net.0.proj.weight",
+        "transformer_blocks.49.img_mlp.net.2.weight",
+        "transformer_blocks.49.img_mod.1.weight",
+        "transformer_blocks.49.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.49.txt_mlp.net.2.weight",
+        "transformer_blocks.49.txt_mod.1.weight",
+        "transformer_blocks.50.attn.add_k_proj.weight",
+        "transformer_blocks.50.attn.add_q_proj.weight",
+        "transformer_blocks.50.attn.add_v_proj.weight",
+        "transformer_blocks.50.attn.to_add_out.weight",
+        "transformer_blocks.50.attn.to_k.weight",
+        "transformer_blocks.50.attn.to_out.0.weight",
+        "transformer_blocks.50.attn.to_q.weight",
+        "transformer_blocks.50.attn.to_v.weight",
+        "transformer_blocks.50.img_mlp.net.0.proj.weight",
+        "transformer_blocks.50.img_mlp.net.2.weight",
+        "transformer_blocks.50.img_mod.1.weight",
+        "transformer_blocks.50.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.50.txt_mlp.net.2.weight",
+        "transformer_blocks.50.txt_mod.1.weight",
+        "transformer_blocks.51.attn.add_k_proj.weight",
+        "transformer_blocks.51.attn.add_q_proj.weight",
+        "transformer_blocks.51.attn.add_v_proj.weight",
+        "transformer_blocks.51.attn.to_add_out.weight",
+        "transformer_blocks.51.attn.to_k.weight",
+        "transformer_blocks.51.attn.to_out.0.weight",
+        "transformer_blocks.51.attn.to_q.weight",
+        "transformer_blocks.51.attn.to_v.weight",
+        "transformer_blocks.51.img_mlp.net.0.proj.weight",
+        "transformer_blocks.51.img_mlp.net.2.weight",
+        "transformer_blocks.51.img_mod.1.weight",
+        "transformer_blocks.51.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.51.txt_mlp.net.2.weight",
+        "transformer_blocks.51.txt_mod.1.weight",
+        "transformer_blocks.52.attn.add_k_proj.weight",
+        "transformer_blocks.52.attn.add_q_proj.weight",
+        "transformer_blocks.52.attn.add_v_proj.weight",
+        "transformer_blocks.52.attn.to_add_out.weight",
+        "transformer_blocks.52.attn.to_k.weight",
+        "transformer_blocks.52.attn.to_out.0.weight",
+        "transformer_blocks.52.attn.to_q.weight",
+        "transformer_blocks.52.attn.to_v.weight",
+        "transformer_blocks.52.img_mlp.net.0.proj.weight",
+        "transformer_blocks.52.img_mlp.net.2.weight",
+        "transformer_blocks.52.img_mod.1.weight",
+        "transformer_blocks.52.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.52.txt_mlp.net.2.weight",
+        "transformer_blocks.52.txt_mod.1.weight",
+        "transformer_blocks.53.attn.add_k_proj.weight",
+        "transformer_blocks.53.attn.add_q_proj.weight",
+        "transformer_blocks.53.attn.add_v_proj.weight",
+        "transformer_blocks.53.attn.to_add_out.weight",
+        "transformer_blocks.53.attn.to_k.weight",
+        "transformer_blocks.53.attn.to_out.0.weight",
+        "transformer_blocks.53.attn.to_q.weight",
+        "transformer_blocks.53.attn.to_v.weight",
+        "transformer_blocks.53.img_mlp.net.0.proj.weight",
+        "transformer_blocks.53.img_mlp.net.2.weight",
+        "transformer_blocks.53.img_mod.1.weight",
+        "transformer_blocks.53.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.53.txt_mlp.net.2.weight",
+        "transformer_blocks.53.txt_mod.1.weight",
+        "transformer_blocks.54.attn.add_k_proj.weight",
+        "transformer_blocks.54.attn.add_q_proj.weight",
+        "transformer_blocks.54.attn.add_v_proj.weight",
+        "transformer_blocks.54.attn.to_add_out.weight",
+        "transformer_blocks.54.attn.to_k.weight",
+        "transformer_blocks.54.attn.to_out.0.weight",
+        "transformer_blocks.54.attn.to_q.weight",
+        "transformer_blocks.54.attn.to_v.weight",
+        "transformer_blocks.54.img_mlp.net.0.proj.weight",
+        "transformer_blocks.54.img_mlp.net.2.weight",
+        "transformer_blocks.54.img_mod.1.weight",
+        "transformer_blocks.54.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.54.txt_mlp.net.2.weight",
+        "transformer_blocks.54.txt_mod.1.weight",
+        "transformer_blocks.55.attn.add_k_proj.weight",
+        "transformer_blocks.55.attn.add_q_proj.weight",
+        "transformer_blocks.55.attn.add_v_proj.weight",
+        "transformer_blocks.55.attn.to_add_out.weight",
+        "transformer_blocks.55.attn.to_k.weight",
+        "transformer_blocks.55.attn.to_out.0.weight",
+        "transformer_blocks.55.attn.to_q.weight",
+        "transformer_blocks.55.attn.to_v.weight",
+        "transformer_blocks.55.img_mlp.net.0.proj.weight",
+        "transformer_blocks.55.img_mlp.net.2.weight",
+        "transformer_blocks.55.img_mod.1.weight",
+        "transformer_blocks.55.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.55.txt_mlp.net.2.weight",
+        "transformer_blocks.55.txt_mod.1.weight",
+        "transformer_blocks.56.attn.add_k_proj.weight",
+        "transformer_blocks.56.attn.add_q_proj.weight",
+        "transformer_blocks.56.attn.add_v_proj.weight",
+        "transformer_blocks.56.attn.to_add_out.weight",
+        "transformer_blocks.56.attn.to_k.weight",
+        "transformer_blocks.56.attn.to_out.0.weight",
+        "transformer_blocks.56.attn.to_q.weight",
+        "transformer_blocks.56.attn.to_v.weight",
+        "transformer_blocks.56.img_mlp.net.0.proj.weight",
+        "transformer_blocks.56.img_mlp.net.2.weight",
+        "transformer_blocks.56.img_mod.1.weight",
+        "transformer_blocks.56.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.56.txt_mlp.net.2.weight",
+        "transformer_blocks.56.txt_mod.1.weight",
+        "transformer_blocks.57.attn.add_k_proj.weight",
+        "transformer_blocks.57.attn.add_q_proj.weight",
+        "transformer_blocks.57.attn.add_v_proj.weight",
+        "transformer_blocks.57.attn.to_add_out.weight",
+        "transformer_blocks.57.attn.to_k.weight",
+        "transformer_blocks.57.attn.to_out.0.weight",
+        "transformer_blocks.57.attn.to_q.weight",
+        "transformer_blocks.57.attn.to_v.weight",
+        "transformer_blocks.57.img_mlp.net.0.proj.weight",
+        "transformer_blocks.57.img_mlp.net.2.weight",
+        "transformer_blocks.57.img_mod.1.weight",
+        "transformer_blocks.57.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.57.txt_mlp.net.2.weight",
+        "transformer_blocks.57.txt_mod.1.weight",
+        "transformer_blocks.58.attn.add_k_proj.weight",
+        "transformer_blocks.58.attn.add_q_proj.weight",
+        "transformer_blocks.58.attn.add_v_proj.weight",
+        "transformer_blocks.58.attn.to_k.weight",
+        "transformer_blocks.58.attn.to_q.weight",
+        "transformer_blocks.58.attn.to_v.weight",
+        "transformer_blocks.58.img_mod.1.weight",
+        "transformer_blocks.58.attn.to_add_out.weight",
+        "transformer_blocks.58.attn.to_out.0.weight",
+        "transformer_blocks.58.img_mlp.net.0.proj.weight",
+        "transformer_blocks.58.img_mlp.net.2.weight",
+        "transformer_blocks.58.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.58.txt_mlp.net.2.weight",
+        "transformer_blocks.58.txt_mod.1.weight",
+        "transformer_blocks.59.attn.add_k_proj.weight",
+        "transformer_blocks.59.attn.add_q_proj.weight",
+        "transformer_blocks.59.attn.add_v_proj.weight",
+        "transformer_blocks.59.attn.to_add_out.weight",
+        "transformer_blocks.59.attn.to_k.weight",
+        "transformer_blocks.59.attn.to_out.0.weight",
+        "transformer_blocks.59.attn.to_q.weight",
+        "transformer_blocks.59.attn.to_v.weight",
+        "transformer_blocks.59.img_mlp.net.0.proj.weight",
+        "transformer_blocks.59.img_mlp.net.2.weight",
+        "transformer_blocks.59.img_mod.1.weight",
+        "transformer_blocks.59.txt_mlp.net.0.proj.weight",
+        "transformer_blocks.59.txt_mlp.net.2.weight",
+        "transformer_blocks.59.txt_mod.1.weight"
+      ]
+    },
+    "modules_quant_config": {},
+    "modules_to_not_convert": [
+      "norm_out",
+      "proj_out",
+      "transformer_blocks.0.img_mod.1.weight",
+      "img_in",
+      "time_text_embed",
+      "txt_in",
+      "transformer_blocks.0.attn.norm_added_k.weight",
+      "transformer_blocks.0.attn.norm_added_q.weight",
+      "transformer_blocks.0.attn.norm_k.weight",
+      "transformer_blocks.0.attn.norm_q.weight",
+      "transformer_blocks.1.attn.norm_added_k.weight",
+      "transformer_blocks.1.attn.norm_added_q.weight",
+      "transformer_blocks.1.attn.norm_k.weight",
+      "transformer_blocks.1.attn.norm_q.weight",
+      "transformer_blocks.2.attn.norm_added_k.weight",
+      "transformer_blocks.2.attn.norm_added_q.weight",
+      "transformer_blocks.2.attn.norm_k.weight",
+      "transformer_blocks.2.attn.norm_q.weight",
+      "transformer_blocks.3.attn.norm_added_k.weight",
+      "transformer_blocks.3.attn.norm_added_q.weight",
+      "transformer_blocks.3.attn.norm_k.weight",
+      "transformer_blocks.3.attn.norm_q.weight",
+      "transformer_blocks.4.attn.norm_added_k.weight",
+      "transformer_blocks.4.attn.norm_added_q.weight",
+      "transformer_blocks.4.attn.norm_k.weight",
+      "transformer_blocks.4.attn.norm_q.weight",
+      "transformer_blocks.5.attn.norm_added_k.weight",
+      "transformer_blocks.5.attn.norm_added_q.weight",
+      "transformer_blocks.5.attn.norm_k.weight",
+      "transformer_blocks.5.attn.norm_q.weight",
+      "transformer_blocks.6.attn.norm_added_k.weight",
+      "transformer_blocks.6.attn.norm_added_q.weight",
+      "transformer_blocks.6.attn.norm_k.weight",
+      "transformer_blocks.6.attn.norm_q.weight",
+      "transformer_blocks.7.attn.norm_k.weight",
+      "transformer_blocks.7.attn.norm_q.weight",
+      "txt_norm.weight",
+      "transformer_blocks.10.attn.norm_added_k.weight",
+      "transformer_blocks.10.attn.norm_added_q.weight",
+      "transformer_blocks.10.attn.norm_k.weight",
+      "transformer_blocks.10.attn.norm_q.weight",
+      "transformer_blocks.11.attn.norm_added_k.weight",
+      "transformer_blocks.11.attn.norm_added_q.weight",
+      "transformer_blocks.11.attn.norm_k.weight",
+      "transformer_blocks.11.attn.norm_q.weight",
+      "transformer_blocks.12.attn.norm_added_k.weight",
+      "transformer_blocks.12.attn.norm_added_q.weight",
+      "transformer_blocks.12.attn.norm_k.weight",
+      "transformer_blocks.12.attn.norm_q.weight",
+      "transformer_blocks.13.attn.norm_added_k.weight",
+      "transformer_blocks.13.attn.norm_added_q.weight",
+      "transformer_blocks.13.attn.norm_k.weight",
+      "transformer_blocks.13.attn.norm_q.weight",
+      "transformer_blocks.14.attn.norm_added_k.weight",
+      "transformer_blocks.14.attn.norm_added_q.weight",
+      "transformer_blocks.14.attn.norm_k.weight",
+      "transformer_blocks.14.attn.norm_q.weight",
+      "transformer_blocks.7.attn.norm_added_k.weight",
+      "transformer_blocks.7.attn.norm_added_q.weight",
+      "transformer_blocks.8.attn.norm_added_k.weight",
+      "transformer_blocks.8.attn.norm_added_q.weight",
+      "transformer_blocks.8.attn.norm_k.weight",
+      "transformer_blocks.8.attn.norm_q.weight",
+      "transformer_blocks.9.attn.norm_added_k.weight",
+      "transformer_blocks.9.attn.norm_added_q.weight",
+      "transformer_blocks.9.attn.norm_k.weight",
+      "transformer_blocks.9.attn.norm_q.weight",
+      "transformer_blocks.15.attn.norm_added_k.weight",
+      "transformer_blocks.15.attn.norm_added_q.weight",
+      "transformer_blocks.15.attn.norm_k.weight",
+      "transformer_blocks.15.attn.norm_q.weight",
+      "transformer_blocks.16.attn.norm_added_k.weight",
+      "transformer_blocks.16.attn.norm_added_q.weight",
+      "transformer_blocks.16.attn.norm_k.weight",
+      "transformer_blocks.16.attn.norm_q.weight",
+      "transformer_blocks.17.attn.norm_added_k.weight",
+      "transformer_blocks.17.attn.norm_added_q.weight",
+      "transformer_blocks.17.attn.norm_k.weight",
+      "transformer_blocks.17.attn.norm_q.weight",
+      "transformer_blocks.18.attn.norm_added_k.weight",
+      "transformer_blocks.18.attn.norm_added_q.weight",
+      "transformer_blocks.18.attn.norm_k.weight",
+      "transformer_blocks.18.attn.norm_q.weight",
+      "transformer_blocks.19.attn.norm_added_k.weight",
+      "transformer_blocks.19.attn.norm_added_q.weight",
+      "transformer_blocks.19.attn.norm_k.weight",
+      "transformer_blocks.19.attn.norm_q.weight",
+      "transformer_blocks.20.attn.norm_added_k.weight",
+      "transformer_blocks.20.attn.norm_added_q.weight",
+      "transformer_blocks.20.attn.norm_k.weight",
+      "transformer_blocks.20.attn.norm_q.weight",
+      "transformer_blocks.21.attn.norm_added_k.weight",
+      "transformer_blocks.21.attn.norm_added_q.weight",
+      "transformer_blocks.21.attn.norm_k.weight",
+      "transformer_blocks.21.attn.norm_q.weight",
+      "transformer_blocks.22.attn.norm_added_k.weight",
+      "transformer_blocks.22.attn.norm_added_q.weight",
+      "transformer_blocks.22.attn.norm_k.weight",
+      "transformer_blocks.22.attn.norm_q.weight",
+      "transformer_blocks.23.attn.norm_added_k.weight",
+      "transformer_blocks.23.attn.norm_added_q.weight",
+      "transformer_blocks.23.attn.norm_k.weight",
+      "transformer_blocks.23.attn.norm_q.weight",
+      "transformer_blocks.24.attn.norm_added_k.weight",
+      "transformer_blocks.24.attn.norm_added_q.weight",
+      "transformer_blocks.24.attn.norm_k.weight",
+      "transformer_blocks.24.attn.norm_q.weight",
+      "transformer_blocks.25.attn.norm_added_k.weight",
+      "transformer_blocks.25.attn.norm_added_q.weight",
+      "transformer_blocks.25.attn.norm_k.weight",
+      "transformer_blocks.25.attn.norm_q.weight",
+      "transformer_blocks.26.attn.norm_added_k.weight",
+      "transformer_blocks.26.attn.norm_added_q.weight",
+      "transformer_blocks.26.attn.norm_k.weight",
+      "transformer_blocks.26.attn.norm_q.weight",
+      "transformer_blocks.27.attn.norm_added_k.weight",
+      "transformer_blocks.27.attn.norm_added_q.weight",
+      "transformer_blocks.27.attn.norm_k.weight",
+      "transformer_blocks.27.attn.norm_q.weight",
+      "transformer_blocks.28.attn.norm_added_k.weight",
+      "transformer_blocks.28.attn.norm_added_q.weight",
+      "transformer_blocks.28.attn.norm_k.weight",
+      "transformer_blocks.28.attn.norm_q.weight",
+      "transformer_blocks.29.attn.norm_k.weight",
+      "transformer_blocks.29.attn.norm_q.weight",
+      "transformer_blocks.29.attn.norm_added_k.weight",
+      "transformer_blocks.29.attn.norm_added_q.weight",
+      "transformer_blocks.30.attn.norm_added_k.weight",
+      "transformer_blocks.30.attn.norm_added_q.weight",
+      "transformer_blocks.30.attn.norm_k.weight",
+      "transformer_blocks.30.attn.norm_q.weight",
+      "transformer_blocks.31.attn.norm_added_k.weight",
+      "transformer_blocks.31.attn.norm_added_q.weight",
+      "transformer_blocks.31.attn.norm_k.weight",
+      "transformer_blocks.31.attn.norm_q.weight",
+      "transformer_blocks.32.attn.norm_added_k.weight",
+      "transformer_blocks.32.attn.norm_added_q.weight",
+      "transformer_blocks.32.attn.norm_k.weight",
+      "transformer_blocks.32.attn.norm_q.weight",
+      "transformer_blocks.33.attn.norm_added_k.weight",
+      "transformer_blocks.33.attn.norm_added_q.weight",
+      "transformer_blocks.33.attn.norm_k.weight",
+      "transformer_blocks.33.attn.norm_q.weight",
+      "transformer_blocks.34.attn.norm_added_k.weight",
+      "transformer_blocks.34.attn.norm_added_q.weight",
+      "transformer_blocks.34.attn.norm_k.weight",
+      "transformer_blocks.34.attn.norm_q.weight",
+      "transformer_blocks.35.attn.norm_added_k.weight",
+      "transformer_blocks.35.attn.norm_added_q.weight",
+      "transformer_blocks.35.attn.norm_k.weight",
+      "transformer_blocks.35.attn.norm_q.weight",
+      "transformer_blocks.36.attn.norm_added_k.weight",
+      "transformer_blocks.36.attn.norm_added_q.weight",
+      "transformer_blocks.36.attn.norm_k.weight",
+      "transformer_blocks.36.attn.norm_q.weight",
+      "transformer_blocks.37.attn.norm_added_k.weight",
+      "transformer_blocks.37.attn.norm_added_q.weight",
+      "transformer_blocks.37.attn.norm_k.weight",
+      "transformer_blocks.37.attn.norm_q.weight",
+      "transformer_blocks.38.attn.norm_added_k.weight",
+      "transformer_blocks.38.attn.norm_added_q.weight",
+      "transformer_blocks.38.attn.norm_k.weight",
+      "transformer_blocks.38.attn.norm_q.weight",
+      "transformer_blocks.39.attn.norm_added_k.weight",
+      "transformer_blocks.39.attn.norm_added_q.weight",
+      "transformer_blocks.39.attn.norm_k.weight",
+      "transformer_blocks.39.attn.norm_q.weight",
+      "transformer_blocks.40.attn.norm_added_k.weight",
+      "transformer_blocks.40.attn.norm_added_q.weight",
+      "transformer_blocks.40.attn.norm_k.weight",
+      "transformer_blocks.40.attn.norm_q.weight",
+      "transformer_blocks.41.attn.norm_added_k.weight",
+      "transformer_blocks.41.attn.norm_added_q.weight",
+      "transformer_blocks.41.attn.norm_k.weight",
+      "transformer_blocks.41.attn.norm_q.weight",
+      "transformer_blocks.42.attn.norm_added_k.weight",
+      "transformer_blocks.42.attn.norm_added_q.weight",
+      "transformer_blocks.42.attn.norm_k.weight",
+      "transformer_blocks.42.attn.norm_q.weight",
+      "transformer_blocks.43.attn.norm_added_k.weight",
+      "transformer_blocks.43.attn.norm_added_q.weight",
+      "transformer_blocks.43.attn.norm_k.weight",
+      "transformer_blocks.43.attn.norm_q.weight",
+      "transformer_blocks.44.attn.norm_added_k.weight",
+      "transformer_blocks.44.attn.norm_added_q.weight",
+      "transformer_blocks.44.attn.norm_k.weight",
+      "transformer_blocks.44.attn.norm_q.weight",
+      "transformer_blocks.45.attn.norm_added_k.weight",
+      "transformer_blocks.45.attn.norm_added_q.weight",
+      "transformer_blocks.45.attn.norm_k.weight",
+      "transformer_blocks.45.attn.norm_q.weight",
+      "transformer_blocks.46.attn.norm_added_k.weight",
+      "transformer_blocks.46.attn.norm_added_q.weight",
+      "transformer_blocks.46.attn.norm_k.weight",
+      "transformer_blocks.46.attn.norm_q.weight",
+      "transformer_blocks.47.attn.norm_added_k.weight",
+      "transformer_blocks.47.attn.norm_added_q.weight",
+      "transformer_blocks.47.attn.norm_k.weight",
+      "transformer_blocks.47.attn.norm_q.weight",
+      "transformer_blocks.48.attn.norm_added_k.weight",
+      "transformer_blocks.48.attn.norm_added_q.weight",
+      "transformer_blocks.48.attn.norm_k.weight",
+      "transformer_blocks.48.attn.norm_q.weight",
+      "transformer_blocks.49.attn.norm_added_k.weight",
+      "transformer_blocks.49.attn.norm_added_q.weight",
+      "transformer_blocks.49.attn.norm_k.weight",
+      "transformer_blocks.49.attn.norm_q.weight",
+      "transformer_blocks.50.attn.norm_added_k.weight",
+      "transformer_blocks.50.attn.norm_added_q.weight",
+      "transformer_blocks.50.attn.norm_k.weight",
+      "transformer_blocks.50.attn.norm_q.weight",
+      "transformer_blocks.51.attn.norm_added_k.weight",
+      "transformer_blocks.51.attn.norm_added_q.weight",
+      "transformer_blocks.51.attn.norm_k.weight",
+      "transformer_blocks.51.attn.norm_q.weight",
+      "transformer_blocks.52.attn.norm_added_k.weight",
+      "transformer_blocks.52.attn.norm_added_q.weight",
+      "transformer_blocks.52.attn.norm_k.weight",
+      "transformer_blocks.52.attn.norm_q.weight",
+      "transformer_blocks.53.attn.norm_added_k.weight",
+      "transformer_blocks.53.attn.norm_added_q.weight",
+      "transformer_blocks.53.attn.norm_k.weight",
+      "transformer_blocks.53.attn.norm_q.weight",
+      "transformer_blocks.54.attn.norm_added_k.weight",
+      "transformer_blocks.54.attn.norm_added_q.weight",
+      "transformer_blocks.54.attn.norm_k.weight",
+      "transformer_blocks.54.attn.norm_q.weight",
+      "transformer_blocks.55.attn.norm_added_k.weight",
+      "transformer_blocks.55.attn.norm_added_q.weight",
+      "transformer_blocks.55.attn.norm_k.weight",
+      "transformer_blocks.55.attn.norm_q.weight",
+      "transformer_blocks.56.attn.norm_added_k.weight",
+      "transformer_blocks.56.attn.norm_added_q.weight",
+      "transformer_blocks.56.attn.norm_k.weight",
+      "transformer_blocks.56.attn.norm_q.weight",
+      "transformer_blocks.57.attn.norm_added_k.weight",
+      "transformer_blocks.57.attn.norm_added_q.weight",
+      "transformer_blocks.57.attn.norm_k.weight",
+      "transformer_blocks.57.attn.norm_q.weight",
+      "transformer_blocks.58.attn.norm_k.weight",
+      "transformer_blocks.58.attn.norm_q.weight",
+      "transformer_blocks.58.attn.norm_added_k.weight",
+      "transformer_blocks.58.attn.norm_added_q.weight",
+      "transformer_blocks.59.attn.norm_added_k.weight",
+      "transformer_blocks.59.attn.norm_added_q.weight",
+      "transformer_blocks.59.attn.norm_k.weight",
+      "transformer_blocks.59.attn.norm_q.weight"
+    ],
+    "modules_to_not_use_matmul": [],
+    "non_blocking": false,
+    "quant_conv": false,
+    "quant_embedding": false,
+    "quant_method": "sdnq",
+    "quantization_device": null,
+    "quantized_matmul_dtype": null,
+    "return_device": null,
+    "sdnq_version": "0.2.0",
+    "svd_rank": 32,
+    "svd_steps": 8,
+    "use_dynamic_quantization": true,
+    "use_grad_ckpt": true,
+    "use_hadamard": true,
+    "use_quantized_matmul": false,
+    "use_quantized_matmul_conv": false,
+    "use_static_quantization": true,
+    "use_stochastic_rounding": false,
+    "use_svd": false,
+    "weights_dtype": "uint4"
+  },
+  "use_additional_t_cond": false,
+  "use_layer3d_rope": false,
+  "zero_cond_t": false
+}

transformer/diffusion_pytorch_model-00001-of-00002.safetensors

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

transformer/diffusion_pytorch_model-00002-of-00002.safetensors

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

vae/config.json

@@ -0,0 +1,56 @@
+{
+  "_class_name": "AutoencoderKLQwenImage",
+  "_diffusers_version": "0.36.0.dev0",
+  "attn_scales": [],
+  "base_dim": 96,
+  "dim_mult": [
+    1,
+    2,
+    4,
+    4
+  ],
+  "dropout": 0.0,
+  "latents_mean": [
+    -0.7571,
+    -0.7089,
+    -0.9113,
+    0.1075,
+    -0.1745,
+    0.9653,
+    -0.1517,
+    1.5508,
+    0.4134,
+    -0.0715,
+    0.5517,
+    -0.3632,
+    -0.1922,
+    -0.9497,
+    0.2503,
+    -0.2921
+  ],
+  "latents_std": [
+    2.8184,
+    1.4541,
+    2.3275,
+    2.6558,
+    1.2196,
+    1.7708,
+    2.6052,
+    2.0743,
+    3.2687,
+    2.1526,
+    2.8652,
+    1.5579,
+    1.6382,
+    1.1253,
+    2.8251,
+    1.916
+  ],
+  "num_res_blocks": 2,
+  "temperal_downsample": [
+    false,
+    true,
+    true
+  ],
+  "z_dim": 16
+}

vae/diffusion_pytorch_model.safetensors

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