UniMate and Flan-T5-base encoder packed for WebGPU, with eight skeletons

LICENSE

@@ -0,0 +1,12 @@
+This repository repackages two models for in-browser inference. Each keeps its original license.
+
+- unimate/ and skeletons/: derived from the UniMate pretrained checkpoints by
+  Linzhan Mou et al. (https://huggingface.co/Linzhan/UniMate), licensed under
+  the Creative Commons Attribution-NonCommercial 4.0 International License
+  (CC BY-NC 4.0). The full notice and license text are in LICENSE-UniMate.
+  Changes are listed in README.md under "Changes from the original".
+  Non-commercial use only.
+
+- t5/: the encoder of google/flan-t5-base (https://huggingface.co/google/flan-t5-base),
+  licensed under the Apache License 2.0
+  (https://www.apache.org/licenses/LICENSE-2.0). Repacked as float32 with some projections concatenated; no values changed.

LICENSE-UniMate

@@ -0,0 +1,425 @@
+UniMate pretrained checkpoints
+Copyright (c) 2026 Linzhan Mou
+
+The model weights and other files in this repository are licensed under the
+Creative Commons Attribution-NonCommercial 4.0 International License
+(CC BY-NC 4.0), reproduced below. They are released for non-commercial
+research use.
+
+This license covers only the rights of the UniMate authors. It grants no
+rights in the third-party data the models were trained on (Truebones ZOO,
+Mixamo, Objaverse-XL), which remain under their own terms; see the
+"License" section of README.md.
+
+The UniMate code (https://github.com/Friedrich-M/UniMate) is released
+separately under the MIT License.
+
+
+Attribution-NonCommercial 4.0 International
+
+=======================================================================
+
+Creative Commons Corporation ("Creative Commons") is not a law firm and
+does not provide legal services or legal advice. Distribution of
+Creative Commons public licenses does not create a lawyer-client or
+other relationship. Creative Commons makes its licenses and related
+information available on an "as-is" basis. Creative Commons gives no
+warranties regarding its licenses, any material licensed under their
+terms and conditions, or any related information. Creative Commons
+disclaims all liability for damages resulting from their use to the
+fullest extent possible.
+
+Using Creative Commons Public Licenses
+
+Creative Commons public licenses provide a standard set of terms and
+conditions that creators and other rights holders may use to share
+original works of authorship and other material subject to copyright
+and certain other rights specified in the public license below. The
+following considerations are for informational purposes only, are not
+exhaustive, and do not form part of our licenses.
+
+     Considerations for licensors: Our public licenses are
+     intended for use by those authorized to give the public
+     permission to use material in ways otherwise restricted by
+     copyright and certain other rights. Our licenses are
+     irrevocable. Licensors should read and understand the terms
+     and conditions of the license they choose before applying it.
+     Licensors should also secure all rights necessary before
+     applying our licenses so that the public can reuse the
+     material as expected. Licensors should clearly mark any
+     material not subject to the license. This includes other CC-
+     licensed material, or material used under an exception or
+     limitation to copyright. More considerations for licensors:
+    wiki.creativecommons.org/Considerations_for_licensors
+
+     Considerations for the public: By using one of our public
+     licenses, a licensor grants the public permission to use the
+     licensed material under specified terms and conditions. If
+     the licensor's permission is not necessary for any reason--for
+     example, because of any applicable exception or limitation to
+     copyright--then that use is not regulated by the license. Our
+     licenses grant only permissions under copyright and certain
+     other rights that a licensor has authority to grant. Use of
+     the licensed material may still be restricted for other
+     reasons, including because others have copyright or other
+     rights in the material. A licensor may make special requests,
+     such as asking that all changes be marked or described.
+     Although not required by our licenses, you are encouraged to
+     respect those requests where reasonable. More considerations
+     for the public:
+    wiki.creativecommons.org/Considerations_for_licensees
+
+=======================================================================
+
+Creative Commons Attribution-NonCommercial 4.0 International Public
+License
+
+By exercising the Licensed Rights (defined below), You accept and agree
+to be bound by the terms and conditions of this Creative Commons
+Attribution-NonCommercial 4.0 International Public License ("Public
+License"). To the extent this Public License may be interpreted as a
+contract, You are granted the Licensed Rights in consideration of Your
+acceptance of these terms and conditions, and the Licensor grants You
+such rights in consideration of benefits the Licensor receives from
+making the Licensed Material available under these terms and
+conditions.
+
+
+Section 1 -- Definitions.
+
+  a. Adapted Material means material subject to Copyright and Similar
+     Rights that is derived from or based upon the Licensed Material
+     and in which the Licensed Material is translated, altered,
+     arranged, transformed, or otherwise modified in a manner requiring
+     permission under the Copyright and Similar Rights held by the
+     Licensor. For purposes of this Public License, where the Licensed
+     Material is a musical work, performance, or sound recording,
+     Adapted Material is always produced where the Licensed Material is
+     synched in timed relation with a moving image.
+
+  b. Adapter's License means the license You apply to Your Copyright
+     and Similar Rights in Your contributions to Adapted Material in
+     accordance with the terms and conditions of this Public License.
+
+  c. Copyright and Similar Rights means copyright and/or similar rights
+     closely related to copyright including, without limitation,
+     performance, broadcast, sound recording, and Sui Generis Database
+     Rights, without regard to how the rights are labeled or
+     categorized. For purposes of this Public License, the rights
+     specified in Section 2(b)(1)-(2) are not Copyright and Similar
+     Rights.
+  d. Effective Technological Measures means those measures that, in the
+     absence of proper authority, may not be circumvented under laws
+     fulfilling obligations under Article 11 of the WIPO Copyright
+     Treaty adopted on December 20, 1996, and/or similar international
+     agreements.
+
+  e. Exceptions and Limitations means fair use, fair dealing, and/or
+     any other exception or limitation to Copyright and Similar Rights
+     that applies to Your use of the Licensed Material.
+
+  f. Licensed Material means the artistic or literary work, database,
+     or other material to which the Licensor applied this Public
+     License.
+
+  g. Licensed Rights means the rights granted to You subject to the
+     terms and conditions of this Public License, which are limited to
+     all Copyright and Similar Rights that apply to Your use of the
+     Licensed Material and that the Licensor has authority to license.
+
+  h. Licensor means the individual(s) or entity(ies) granting rights
+     under this Public License.
+
+  i. NonCommercial means not primarily intended for or directed towards
+     commercial advantage or monetary compensation. For purposes of
+     this Public License, the exchange of the Licensed Material for
+     other material subject to Copyright and Similar Rights by digital
+     file-sharing or similar means is NonCommercial provided there is
+     no payment of monetary compensation in connection with the
+     exchange.
+
+  j. Share means to provide material to the public by any means or
+     process that requires permission under the Licensed Rights, such
+     as reproduction, public display, public performance, distribution,
+     dissemination, communication, or importation, and to make material
+     available to the public including in ways that members of the
+     public may access the material from a place and at a time
+     individually chosen by them.
+
+  k. Sui Generis Database Rights means rights other than copyright
+     resulting from Directive 96/9/EC of the European Parliament and of
+     the Council of 11 March 1996 on the legal protection of databases,
+     as amended and/or succeeded, as well as other essentially
+     equivalent rights anywhere in the world.
+
+  l. You means the individual or entity exercising the Licensed Rights
+     under this Public License. Your has a corresponding meaning.
+
+
+Section 2 -- Scope.
+
+  a. License grant.
+
+       1. Subject to the terms and conditions of this Public License,
+          the Licensor hereby grants You a worldwide, royalty-free,
+          non-sublicensable, non-exclusive, irrevocable license to
+          exercise the Licensed Rights in the Licensed Material to:
+
+            a. reproduce and Share the Licensed Material, in whole or
+               in part, for NonCommercial purposes only; and
+
+            b. produce, reproduce, and Share Adapted Material for
+               NonCommercial purposes only.
+
+       2. Exceptions and Limitations. For the avoidance of doubt, where
+          Exceptions and Limitations apply to Your use, this Public
+          License does not apply, and You do not need to comply with
+          its terms and conditions.
+
+       3. Term. The term of this Public License is specified in Section
+          6(a).
+
+       4. Media and formats; technical modifications allowed. The
+          Licensor authorizes You to exercise the Licensed Rights in
+          all media and formats whether now known or hereafter created,
+          and to make technical modifications necessary to do so. The
+          Licensor waives and/or agrees not to assert any right or
+          authority to forbid You from making technical modifications
+          necessary to exercise the Licensed Rights, including
+          technical modifications necessary to circumvent Effective
+          Technological Measures. For purposes of this Public License,
+          simply making modifications authorized by this Section 2(a)
+          (4) never produces Adapted Material.
+
+       5. Downstream recipients.
+
+            a. Offer from the Licensor -- Licensed Material. Every
+               recipient of the Licensed Material automatically
+               receives an offer from the Licensor to exercise the
+               Licensed Rights under the terms and conditions of this
+               Public License.
+
+            b. No downstream restrictions. You may not offer or impose
+               any additional or different terms or conditions on, or
+               apply any Effective Technological Measures to, the
+               Licensed Material if doing so restricts exercise of the
+               Licensed Rights by any recipient of the Licensed
+               Material.
+
+       6. No endorsement. Nothing in this Public License constitutes or
+          may be construed as permission to assert or imply that You
+          are, or that Your use of the Licensed Material is, connected
+          with, or sponsored, endorsed, or granted official status by,
+          the Licensor or others designated to receive attribution as
+          provided in Section 3(a)(1)(A)(i).
+
+  b. Other rights.
+
+       1. Moral rights, such as the right of integrity, are not
+          licensed under this Public License, nor are publicity,
+          privacy, and/or other similar personality rights; however, to
+          the extent possible, the Licensor waives and/or agrees not to
+          assert any such rights held by the Licensor to the limited
+          extent necessary to allow You to exercise the Licensed
+          Rights, but not otherwise.
+
+       2. Patent and trademark rights are not licensed under this
+          Public License.
+
+       3. To the extent possible, the Licensor waives any right to
+          collect royalties from You for the exercise of the Licensed
+          Rights, whether directly or through a collecting society
+          under any voluntary or waivable statutory or compulsory
+          licensing scheme. In all other cases the Licensor expressly
+          reserves any right to collect such royalties, including when
+          the Licensed Material is used other than for NonCommercial
+          purposes.
+
+
+Section 3 -- License Conditions.
+
+Your exercise of the Licensed Rights is expressly made subject to the
+following conditions.
+
+  a. Attribution.
+
+       1. If You Share the Licensed Material (including in modified
+          form), You must:
+
+            a. retain the following if it is supplied by the Licensor
+               with the Licensed Material:
+
+                 i. identification of the creator(s) of the Licensed
+                    Material and any others designated to receive
+                    attribution, in any reasonable manner requested by
+                    the Licensor (including by pseudonym if
+                    designated);
+
+                ii. a copyright notice;
+
+               iii. a notice that refers to this Public License;
+
+                iv. a notice that refers to the disclaimer of
+                    warranties;
+
+                 v. a URI or hyperlink to the Licensed Material to the
+                    extent reasonably practicable;
+
+            b. indicate if You modified the Licensed Material and
+               retain an indication of any previous modifications; and
+
+            c. indicate the Licensed Material is licensed under this
+               Public License, and include the text of, or the URI or
+               hyperlink to, this Public License.
+
+       2. You may satisfy the conditions in Section 3(a)(1) in any
+          reasonable manner based on the medium, means, and context in
+          which You Share the Licensed Material. For example, it may be
+          reasonable to satisfy the conditions by providing a URI or
+          hyperlink to a resource that includes the required
+          information.
+
+       3. If requested by the Licensor, You must remove any of the
+          information required by Section 3(a)(1)(A) to the extent
+          reasonably practicable.
+
+       4. If You Share Adapted Material You produce, the Adapter's
+          License You apply must not prevent recipients of the Adapted
+          Material from complying with this Public License.
+
+
+Section 4 -- Sui Generis Database Rights.
+
+Where the Licensed Rights include Sui Generis Database Rights that
+apply to Your use of the Licensed Material:
+
+  a. for the avoidance of doubt, Section 2(a)(1) grants You the right
+     to extract, reuse, reproduce, and Share all or a substantial
+     portion of the contents of the database for NonCommercial purposes
+     only;
+
+  b. if You include all or a substantial portion of the database
+     contents in a database in which You have Sui Generis Database
+     Rights, then the database in which You have Sui Generis Database
+     Rights (but not its individual contents) is Adapted Material; and
+
+  c. You must comply with the conditions in Section 3(a) if You Share
+     all or a substantial portion of the contents of the database.
+
+For the avoidance of doubt, this Section 4 supplements and does not
+replace Your obligations under this Public License where the Licensed
+Rights include other Copyright and Similar Rights.
+
+
+Section 5 -- Disclaimer of Warranties and Limitation of Liability.
+
+  a. UNLESS OTHERWISE SEPARATELY UNDERTAKEN BY THE LICENSOR, TO THE
+     EXTENT POSSIBLE, THE LICENSOR OFFERS THE LICENSED MATERIAL AS-IS
+     AND AS-AVAILABLE, AND MAKES NO REPRESENTATIONS OR WARRANTIES OF
+     ANY KIND CONCERNING THE LICENSED MATERIAL, WHETHER EXPRESS,
+     IMPLIED, STATUTORY, OR OTHER. THIS INCLUDES, WITHOUT LIMITATION,
+     WARRANTIES OF TITLE, MERCHANTABILITY, FITNESS FOR A PARTICULAR
+     PURPOSE, NON-INFRINGEMENT, ABSENCE OF LATENT OR OTHER DEFECTS,
+     ACCURACY, OR THE PRESENCE OR ABSENCE OF ERRORS, WHETHER OR NOT
+     KNOWN OR DISCOVERABLE. WHERE DISCLAIMERS OF WARRANTIES ARE NOT
+     ALLOWED IN FULL OR IN PART, THIS DISCLAIMER MAY NOT APPLY TO YOU.
+
+  b. TO THE EXTENT POSSIBLE, IN NO EVENT WILL THE LICENSOR BE LIABLE
+     TO YOU ON ANY LEGAL THEORY (INCLUDING, WITHOUT LIMITATION,
+     NEGLIGENCE) OR OTHERWISE FOR ANY DIRECT, SPECIAL, INDIRECT,
+     INCIDENTAL, CONSEQUENTIAL, PUNITIVE, EXEMPLARY, OR OTHER LOSSES,
+     COSTS, EXPENSES, OR DAMAGES ARISING OUT OF THIS PUBLIC LICENSE OR
+     USE OF THE LICENSED MATERIAL, EVEN IF THE LICENSOR HAS BEEN
+     ADVISED OF THE POSSIBILITY OF SUCH LOSSES, COSTS, EXPENSES, OR
+     DAMAGES. WHERE A LIMITATION OF LIABILITY IS NOT ALLOWED IN FULL OR
+     IN PART, THIS LIMITATION MAY NOT APPLY TO YOU.
+
+  c. The disclaimer of warranties and limitation of liability provided
+     above shall be interpreted in a manner that, to the extent
+     possible, most closely approximates an absolute disclaimer and
+     waiver of all liability.
+
+
+Section 6 -- Term and Termination.
+
+  a. This Public License applies for the term of the Copyright and
+     Similar Rights licensed here. However, if You fail to comply with
+     this Public License, then Your rights under this Public License
+     terminate automatically.
+
+  b. Where Your right to use the Licensed Material has terminated under
+     Section 6(a), it reinstates:
+
+       1. automatically as of the date the violation is cured, provided
+          it is cured within 30 days of Your discovery of the
+          violation; or
+
+       2. upon express reinstatement by the Licensor.
+
+     For the avoidance of doubt, this Section 6(b) does not affect any
+     right the Licensor may have to seek remedies for Your violations
+     of this Public License.
+
+  c. For the avoidance of doubt, the Licensor may also offer the
+     Licensed Material under separate terms or conditions or stop
+     distributing the Licensed Material at any time; however, doing so
+     will not terminate this Public License.
+
+  d. Sections 1, 5, 6, 7, and 8 survive termination of this Public
+     License.
+
+
+Section 7 -- Other Terms and Conditions.
+
+  a. The Licensor shall not be bound by any additional or different
+     terms or conditions communicated by You unless expressly agreed.
+
+  b. Any arrangements, understandings, or agreements regarding the
+     Licensed Material not stated herein are separate from and
+     independent of the terms and conditions of this Public License.
+
+
+Section 8 -- Interpretation.
+
+  a. For the avoidance of doubt, this Public License does not, and
+     shall not be interpreted to, reduce, limit, restrict, or impose
+     conditions on any use of the Licensed Material that could lawfully
+     be made without permission under this Public License.
+
+  b. To the extent possible, if any provision of this Public License is
+     deemed unenforceable, it shall be automatically reformed to the
+     minimum extent necessary to make it enforceable. If the provision
+     cannot be reformed, it shall be severed from this Public License
+     without affecting the enforceability of the remaining terms and
+     conditions.
+
+  c. No term or condition of this Public License will be waived and no
+     failure to comply consented to unless expressly agreed to by the
+     Licensor.
+
+  d. Nothing in this Public License constitutes or may be interpreted
+     as a limitation upon, or waiver of, any privileges and immunities
+     that apply to the Licensor or You, including from the legal
+     processes of any jurisdiction or authority.
+
+=======================================================================
+
+Creative Commons is not a party to its public
+licenses. Notwithstanding, Creative Commons may elect to apply one of
+its public licenses to material it publishes and in those instances
+will be considered the “Licensor.” The text of the Creative Commons
+public licenses is dedicated to the public domain under the CC0 Public
+Domain Dedication. Except for the limited purpose of indicating that
+material is shared under a Creative Commons public license or as
+otherwise permitted by the Creative Commons policies published at
+creativecommons.org/policies, Creative Commons does not authorize the
+use of the trademark "Creative Commons" or any other trademark or logo
+of Creative Commons without its prior written consent including,
+without limitation, in connection with any unauthorized modifications
+to any of its public licenses or any other arrangements,
+understandings, or agreements concerning use of licensed material. For
+the avoidance of doubt, this paragraph does not form part of the
+public licenses.
+
+Creative Commons may be contacted at creativecommons.org.
+

README.md

@@ -0,0 +1,42 @@
+---
+license: cc-by-nc-4.0
+base_model: Linzhan/UniMate
+library_name: kaminos
+tags:
+- webgpu
+- browser
+- motion-generation
+- text-to-motion
+- skeletal-animation
+- unimate
+---
+
+# UniMate for WebGPU
+
+The [UniMate](https://huggingface.co/Linzhan/UniMate) text-to-motion model and its Flan-T5-base text encoder, packed for generating skeletal animation entirely in the browser with the [Kaminos](https://github.com/lyonsno/kaminos) WebGPU inference kit. Give it a skeleton and a prompt such as "A dog runs forward.", and the visitor's own GPU generates a two-second, 30 fps animation for that body. No server-side compute is involved.
+
+**[Try it in your browser](https://lyonsno.github.io/kaminos/inference-kit/unimate/)** · [Source](https://github.com/lyonsno/kaminos/tree/main/models/unimate)
+
+The page downloads about 740 MB once and caches it. A clip takes about 7 seconds on an Apple M4 Max, while the page's 3D scene keeps drawing. The browser port matches the reference PyTorch implementation: one network evaluation agrees to a relative error of about 1.5 × 10⁻⁶, and a whole generation from the same starting noise reproduces the reference animation.
+
+## Contents
+
+| Folder | Component | Format | Size |
+| --- | --- | --- | --- |
+| `unimate/` | UniMate v3 motion transformer (UniML3D, 150k steps, EMA weights), 74.5M parameters | float32 | 298 MB |
+| `t5/` | Flan-T5-base encoder and its SentencePiece vocabulary | float32; JSON | 440 MB |
+| `skeletons/` | Skeleton conditioning for eight bodies: the Unitree Go2 robot dog and seven Truebones animals (cat, dog, horse, T-rex, spider, eagle, parrot) | float32 / int32 | under 0.5 MB each |
+
+Each weight folder has a manifest listing every tensor's name, shape, byte offset and length, and the SHA-256 of the packed file. `skeletons/index.json` lists the bodies with their joint counts and suggested prompts. Each skeleton holds the model's conditioning for that body: rest pose, hierarchy, joint-name embeddings, graph and spectral features, plus the feature normalization statistics.
+
+## Changes from the original
+
+- The UniMate weights are the released checkpoint with its EMA weights applied (as UniMate's own sampler loads them), copied unchanged into one 256-byte-aligned float32 file. Loaded back into the reference model, the packed file reproduces its outputs bit for bit.
+- The text encoder is the encoder half of Flan-T5-base. Each layer's query, key and value projections are concatenated into one matrix, as are the two gated feed-forward input projections; no values are changed.
+- Skeleton conditioning was produced by UniMate's own preprocessing from its released Truebones features and the Go2 example asset.
+
+## License
+
+The UniMate weights and the skeleton conditioning derived from them are licensed under **CC BY-NC 4.0**, the license of the original checkpoints, for non-commercial use only. See `LICENSE-UniMate`. That license covers only the rights of the UniMate authors; the data UniMate was trained on keeps its own terms. Flan-T5-base is by Google, under the Apache License 2.0.
+
+UniMate is by Linzhan Mou, Jiahui Lei, Zhiyang Dou, Chenyue Cai, Chaoyue Song, Adam Finkelstein and Szymon Rusinkiewicz: [paper](https://arxiv.org/abs/2609.05415), [code](https://github.com/Friedrich-M/UniMate), [project page](https://linzhanmou.com/unimate/).

skeletons/cat/demo.bin

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

skeletons/cat/demo.json

@@ -0,0 +1,303 @@
+{
+ "schema": "kaminos.unimate.demo-bundle.v0",
+ "skeleton": {
+  "name": "Cat",
+  "joints": 43,
+  "parents": [
+   -1,
+   0,
+   0,
+   1,
+   1,
+   1,
+   2,
+   3,
+   4,
+   5,
+   6,
+   7,
+   8,
+   9,
+   10,
+   11,
+   11,
+   11,
+   12,
+   13,
+   15,
+   16,
+   17,
+   18,
+   19,
+   20,
+   20,
+   20,
+   20,
+   20,
+   20,
+   21,
+   22,
+   23,
+   24,
+   25,
+   31,
+   32,
+   35,
+   36,
+   37,
+   39,
+   40
+  ],
+  "jointNames": [
+   "Bip01_Pelvis",
+   "Bip01_Spine",
+   "BN_Tail_01",
+   "Bip01_Spine1",
+   "Bip01_L_Thigh",
+   "Bip01_R_Thigh",
+   "BN_Tail_02",
+   "Bip01_Spine2",
+   "Bip01_L_Calf",
+   "Bip01_R_Calf",
+   "BN_Tail_03",
+   "Bip01_Neck",
+   "Bip01_L_HorseLink",
+   "Bip01_R_HorseLink",
+   "BN_Tail_04",
+   "Bip01_Neck1",
+   "Bip01_R_Clavicle",
+   "Bip01_L_Clavicle",
+   "Bip01_L_Foot",
+   "Bip01_R_Foot",
+   "Bip01_Head",
+   "Bip01_R_UpperArm",
+   "Bip01_L_UpperArm",
+   "Bip01_L_Toe0",
+   "Bip01_R_Toe0",
+   "BN_Mouth",
+   "BN_Ear_L",
+   "BN_Ear_R",
+   "BN_Beard_R",
+   "BN_Beard_L",
+   "Bip01_HeadNub",
+   "Bip01_R_Forearm",
+   "Bip01_L_Forearm",
+   "Bip01_L_Toe0Nub",
+   "Bip01_R_Toe0Nub",
+   "BN_Thouge_01",
+   "Bip01_R_Hand",
+   "Bip01_L_Hand",
+   "BN_Thouge_02",
+   "Bip01_R_Finger0",
+   "Bip01_L_Finger0",
+   "Bip01_R_Finger0Nub",
+   "Bip01_L_Finger0Nub"
+  ],
+  "fps": 30,
+  "frames": 60
+ },
+ "motionLength": 60,
+ "outputTimes": [
+  0.0,
+  0.020408162847161293,
+  0.040816325694322586,
+  0.06122449040412903,
+  0.08163265138864517,
+  0.10204081237316132,
+  0.12244898080825806,
+  0.1428571343421936,
+  0.16326530277729034,
+  0.18367347121238708,
+  0.20408162474632263,
+  0.22448979318141937,
+  0.2448979616165161,
+  0.26530611515045166,
+  0.2857142686843872,
+  0.30612245202064514,
+  0.3265306055545807,
+  0.34693875908851624,
+  0.36734694242477417,
+  0.3877550959587097,
+  0.40816324949264526,
+  0.4285714328289032,
+  0.44897958636283875,
+  0.4693877398967743,
+  0.4897959232330322,
+  0.5102040767669678,
+  0.5306122303009033,
+  0.5510204434394836,
+  0.5714285969734192,
+  0.5918367505073547,
+  0.6122449040412903,
+  0.6326530575752258,
+  0.6530612111091614,
+  0.6734694242477417,
+  0.6938775777816772,
+  0.7142857313156128,
+  0.7346938848495483,
+  0.7551020383834839,
+  0.7755101919174194,
+  0.795918345451355,
+  0.8163265585899353,
+  0.8367347121238708,
+  0.8571428656578064,
+  0.8775510191917419,
+  0.8979591727256775,
+  0.918367326259613,
+  0.9387755393981934,
+  0.9591836929321289,
+  0.9795918464660645,
+  1.0
+ ],
+ "prompts": [
+  {
+   "text": "A cat walks forward.",
+   "caption": "caption0",
+   "referenceNoise": "noise0",
+   "referencePositions": "reference0"
+  }
+ ],
+ "assetCondSha256": "a9ff66a4a910c9c1e94192f09d813af776f54014dd3eed4929247d14bd0372a8",
+ "sha256": "6422d4fe9c13557bd48e4afda42fe08c0d7ac8da55dabc92b2c6d6316e18c33f",
+ "arrays": [
+  {
+   "name": "tpos",
+   "shape": [
+    43,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 0,
+   "byteLength": 2064
+  },
+  {
+   "name": "tposParents",
+   "shape": [
+    43,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 2304,
+   "byteLength": 2064
+  },
+  {
+   "name": "depths",
+   "shape": [
+    43
+   ],
+   "dtype": "int32",
+   "offset": 4608,
+   "byteLength": 172
+  },
+  {
+   "name": "jointNamesEmb",
+   "shape": [
+    43,
+    768
+   ],
+   "dtype": "float32",
+   "offset": 4864,
+   "byteLength": 132096
+  },
+  {
+   "name": "spectral",
+   "shape": [
+    43,
+    8
+   ],
+   "dtype": "float32",
+   "offset": 136960,
+   "byteLength": 1376
+  },
+  {
+   "name": "graphDist",
+   "shape": [
+    43,
+    43
+   ],
+   "dtype": "int32",
+   "offset": 138496,
+   "byteLength": 7396
+  },
+  {
+   "name": "jointRelations",
+   "shape": [
+    43,
+    43
+   ],
+   "dtype": "int32",
+   "offset": 145920,
+   "byteLength": 7396
+  },
+  {
+   "name": "mean",
+   "shape": [
+    43,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 153344,
+   "byteLength": 2064
+  },
+  {
+   "name": "std",
+   "shape": [
+    43,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 155648,
+   "byteLength": 2064
+  },
+  {
+   "name": "restPositions",
+   "shape": [
+    43,
+    3
+   ],
+   "dtype": "float32",
+   "offset": 157952,
+   "byteLength": 516
+  },
+  {
+   "name": "caption0",
+   "shape": [
+    768
+   ],
+   "dtype": "float32",
+   "offset": 158720,
+   "byteLength": 3072
+  },
+  {
+   "name": "noise0",
+   "shape": [
+    43,
+    12,
+    60
+   ],
+   "dtype": "float32",
+   "offset": 161792,
+   "byteLength": 123840
+  },
+  {
+   "name": "reference0",
+   "shape": [
+    60,
+    43,
+    3
+   ],
+   "dtype": "float32",
+   "offset": 285696,
+   "byteLength": 30960
+  }
+ ],
+ "captureIdentity": {
+  "prompts": [
+   "A cat walks forward."
+  ],
+  "seed": 10,
+  "model_sha256": "f991ddd87262c251f4681d18297a6163ab4f139daa28c86d5ef0a66771df7bc6",
+  "config_sha256": "c8d3b44ee5ea6c5bd71e7b5ad1f4c94019314a71ac2578d59352618b6342374c",
+  "stats_sha256": "0ff2b9ef3ea9a15b024ab0670e55f659478251501e5596f9011d5bca4cf67773"
+ }
+}

skeletons/dog/demo.bin

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

skeletons/dog/demo.json

@@ -0,0 +1,307 @@
+{
+ "schema": "kaminos.unimate.demo-bundle.v0",
+ "skeleton": {
+  "name": "Dog",
+  "joints": 51,
+  "parents": [
+   -1,
+   0,
+   0,
+   0,
+   0,
+   1,
+   2,
+   3,
+   4,
+   5,
+   6,
+   7,
+   8,
+   9,
+   10,
+   11,
+   12,
+   13,
+   14,
+   15,
+   16,
+   17,
+   17,
+   17,
+   18,
+   19,
+   21,
+   22,
+   23,
+   26,
+   27,
+   28,
+   29,
+   29,
+   29,
+   29,
+   29,
+   29,
+   30,
+   31,
+   32,
+   32,
+   33,
+   34,
+   38,
+   39,
+   40,
+   44,
+   45,
+   46,
+   49
+  ],
+  "jointNames": [
+   "Bip01_Pelvis",
+   "Bip01_Spine",
+   "Bip01_L_Thigh",
+   "Bip01_R_Thigh",
+   "Bip01_Spine0_Tail",
+   "Bip01_Spine1",
+   "Bip01_L_Calf",
+   "Bip01_R_Calf",
+   "Bip01_Spine0_Tail1",
+   "Bip01_Spine2",
+   "Bip01_L_Foot",
+   "Bip01_R_Foot",
+   "Bip01_Spine0_Tail2",
+   "Bip01_Spine3",
+   "Bip01_L_Toe0",
+   "Bip01_R_Toe0",
+   "Bip01_Spine0_Tail3",
+   "Bip01_Neck",
+   "Bip01_L_Toe01",
+   "Bip01_R_Toe01",
+   "Bip01_Spine0_Tail4",
+   "Bip01_Neck1",
+   "Bip01_L_Clavicle",
+   "Bip01_R_Clavicle",
+   "Bip01_L_Toe02",
+   "Bip01_R_Toe02",
+   "Bip01_Neck2",
+   "Bip01_L_UpperArm",
+   "Bip01_R_UpperArm",
+   "Bip01_Head",
+   "Bip01_L_Forearm",
+   "Bip01_R_Forearm",
+   "Bip01_Head1_Jaw",
+   "Bip01_Ponytail2",
+   "Bip01_Ponytail1",
+   "EyesBlue_2",
+   "Bip01_Head2_Eyeleds",
+   "Bip01_Head_Muzzle",
+   "Bip01_L_Hand",
+   "Bip01_R_Hand",
+   "_Bip01_Head11_TungeControler",
+   "Bip01_Head_JawNub",
+   "Bip01_Ponytail21",
+   "Bip01_Ponytail11",
+   "Bip01_L_Finger0",
+   "Bip01_R_Finger0",
+   "_Bip01_Ponytail3_tunge",
+   "Bip01_L_Finger01",
+   "Bip01_R_Finger01",
+   "Bip01_Ponytail31",
+   "Bip01_Ponytail32"
+  ],
+  "fps": 30,
+  "frames": 60
+ },
+ "motionLength": 60,
+ "outputTimes": [
+  0.0,
+  0.020408162847161293,
+  0.040816325694322586,
+  0.06122449040412903,
+  0.08163265138864517,
+  0.10204081237316132,
+  0.12244898080825806,
+  0.1428571343421936,
+  0.16326530277729034,
+  0.18367347121238708,
+  0.20408162474632263,
+  0.22448979318141937,
+  0.2448979616165161,
+  0.26530611515045166,
+  0.2857142686843872,
+  0.30612245202064514,
+  0.3265306055545807,
+  0.34693875908851624,
+  0.36734694242477417,
+  0.3877550959587097,
+  0.40816324949264526,
+  0.4285714328289032,
+  0.44897958636283875,
+  0.4693877398967743,
+  0.4897959232330322,
+  0.5102040767669678,
+  0.5306122303009033,
+  0.5510204434394836,
+  0.5714285969734192,
+  0.5918367505073547,
+  0.6122449040412903,
+  0.6326530575752258,
+  0.6530612111091614,
+  0.6734694242477417,
+  0.6938775777816772,
+  0.7142857313156128,
+  0.7346938848495483,
+  0.7551020383834839,
+  0.7755101919174194,
+  0.795918345451355,
+  0.8163265585899353,
+  0.8367347121238708,
+  0.8571428656578064,
+  0.8775510191917419,
+  0.8979591727256775,
+  0.918367326259613,
+  0.9387755393981934,
+  0.9591836929321289,
+  0.9795918464660645,
+  1.0
+ ],
+ "prompts": [
+  {
+   "text": "An animal walks forward.",
+   "caption": "caption0",
+   "referenceNoise": "noise0"
+  }
+ ],
+ "assetCondSha256": "a9ff66a4a910c9c1e94192f09d813af776f54014dd3eed4929247d14bd0372a8",
+ "sha256": "e013d1da8cd0392db4d19a766ef7f60ff1daa69b18df2d0a025da45db9646a74",
+ "arrays": [
+  {
+   "name": "tpos",
+   "shape": [
+    51,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 0,
+   "byteLength": 2448
+  },
+  {
+   "name": "tposParents",
+   "shape": [
+    51,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 2560,
+   "byteLength": 2448
+  },
+  {
+   "name": "depths",
+   "shape": [
+    51
+   ],
+   "dtype": "int32",
+   "offset": 5120,
+   "byteLength": 204
+  },
+  {
+   "name": "jointNamesEmb",
+   "shape": [
+    51,
+    768
+   ],
+   "dtype": "float32",
+   "offset": 5376,
+   "byteLength": 156672
+  },
+  {
+   "name": "spectral",
+   "shape": [
+    51,
+    8
+   ],
+   "dtype": "float32",
+   "offset": 162048,
+   "byteLength": 1632
+  },
+  {
+   "name": "graphDist",
+   "shape": [
+    51,
+    51
+   ],
+   "dtype": "int32",
+   "offset": 163840,
+   "byteLength": 10404
+  },
+  {
+   "name": "jointRelations",
+   "shape": [
+    51,
+    51
+   ],
+   "dtype": "int32",
+   "offset": 174336,
+   "byteLength": 10404
+  },
+  {
+   "name": "mean",
+   "shape": [
+    51,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 184832,
+   "byteLength": 2448
+  },
+  {
+   "name": "std",
+   "shape": [
+    51,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 187392,
+   "byteLength": 2448
+  },
+  {
+   "name": "restPositions",
+   "shape": [
+    51,
+    3
+   ],
+   "dtype": "float32",
+   "offset": 189952,
+   "byteLength": 612
+  },
+  {
+   "name": "caption0",
+   "shape": [
+    768
+   ],
+   "dtype": "float32",
+   "offset": 190720,
+   "byteLength": 3072
+  },
+  {
+   "name": "noise0",
+   "shape": [
+    51,
+    12,
+    60
+   ],
+   "dtype": "float32",
+   "offset": 193792,
+   "byteLength": 146880
+  }
+ ],
+ "captureIdentity": {
+  "prompts": [
+   "An animal walks forward."
+  ],
+  "seed": 10,
+  "model_sha256": "f991ddd87262c251f4681d18297a6163ab4f139daa28c86d5ef0a66771df7bc6",
+  "config_sha256": "c8d3b44ee5ea6c5bd71e7b5ad1f4c94019314a71ac2578d59352618b6342374c",
+  "stats_sha256": "0ff2b9ef3ea9a15b024ab0670e55f659478251501e5596f9011d5bca4cf67773"
+ }
+}

skeletons/eagle/demo.bin

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

skeletons/eagle/demo.json

@@ -0,0 +1,307 @@
+{
+ "schema": "kaminos.unimate.demo-bundle.v0",
+ "skeleton": {
+  "name": "Eagle",
+  "joints": 51,
+  "parents": [
+   -1,
+   0,
+   0,
+   1,
+   1,
+   1,
+   2,
+   2,
+   3,
+   3,
+   3,
+   4,
+   5,
+   6,
+   7,
+   8,
+   9,
+   10,
+   11,
+   12,
+   15,
+   16,
+   17,
+   18,
+   19,
+   20,
+   20,
+   21,
+   21,
+   22,
+   23,
+   23,
+   23,
+   23,
+   23,
+   24,
+   24,
+   24,
+   24,
+   24,
+   25,
+   27,
+   29,
+   30,
+   31,
+   32,
+   33,
+   35,
+   36,
+   37,
+   38
+  ],
+  "jointNames": [
+   "Bip01_Pelvis",
+   "Bip01_Spine",
+   "BN_Tai_01",
+   "Bip01_Spine1",
+   "Bip01_L_Thigh",
+   "Bip01_R_Thigh",
+   "BN_Tai_L_01",
+   "BN_Tai_R_01",
+   "BN_Wing_R_01",
+   "BN_Wing_L_01",
+   "Bip01_Neck",
+   "Bip01_L_Calf",
+   "Bip01_R_Calf",
+   "BN_Tai_L_02",
+   "BN_Tai_R_02",
+   "BN_Wing_R_02",
+   "BN_Wing_L_02",
+   "Bip01_Neck1",
+   "Bip01_L_HorseLink",
+   "Bip01_R_HorseLink",
+   "BN_Wing_R_03",
+   "BN_Wing_L_03",
+   "Bip01_Neck2",
+   "Bip01_L_Foot",
+   "Bip01_R_Foot",
+   "BN_Wing_R_05",
+   "BN_Wing_R_04",
+   "BN_Wing_L_05",
+   "BN_Wing_L_04",
+   "Bip01_Head",
+   "BN_Toe_L_30",
+   "BN_Toe_L_10",
+   "BN_Toe_L_20",
+   "BN_Toe_L_00",
+   "Bip01_L_Toe0",
+   "BN_Toe_R_30",
+   "BN_Toe_R_10",
+   "BN_Toe_R_20",
+   "BN_Toe_R_00",
+   "Bip01_R_Toe0",
+   "BN_Wing_R_06",
+   "BN_Wing_L_06",
+   "BN_Jaw",
+   "BN_Toe_L_31",
+   "BN_Toe_L_11",
+   "BN_Toe_L_21",
+   "BN_Toe_L_01",
+   "BN_Toe_R_31",
+   "BN_Toe_R_11",
+   "BN_Toe_R_21",
+   "BN_Toe_R_01"
+  ],
+  "fps": 30,
+  "frames": 60
+ },
+ "motionLength": 60,
+ "outputTimes": [
+  0.0,
+  0.020408162847161293,
+  0.040816325694322586,
+  0.06122449040412903,
+  0.08163265138864517,
+  0.10204081237316132,
+  0.12244898080825806,
+  0.1428571343421936,
+  0.16326530277729034,
+  0.18367347121238708,
+  0.20408162474632263,
+  0.22448979318141937,
+  0.2448979616165161,
+  0.26530611515045166,
+  0.2857142686843872,
+  0.30612245202064514,
+  0.3265306055545807,
+  0.34693875908851624,
+  0.36734694242477417,
+  0.3877550959587097,
+  0.40816324949264526,
+  0.4285714328289032,
+  0.44897958636283875,
+  0.4693877398967743,
+  0.4897959232330322,
+  0.5102040767669678,
+  0.5306122303009033,
+  0.5510204434394836,
+  0.5714285969734192,
+  0.5918367505073547,
+  0.6122449040412903,
+  0.6326530575752258,
+  0.6530612111091614,
+  0.6734694242477417,
+  0.6938775777816772,
+  0.7142857313156128,
+  0.7346938848495483,
+  0.7551020383834839,
+  0.7755101919174194,
+  0.795918345451355,
+  0.8163265585899353,
+  0.8367347121238708,
+  0.8571428656578064,
+  0.8775510191917419,
+  0.8979591727256775,
+  0.918367326259613,
+  0.9387755393981934,
+  0.9591836929321289,
+  0.9795918464660645,
+  1.0
+ ],
+ "prompts": [
+  {
+   "text": "An animal walks forward.",
+   "caption": "caption0",
+   "referenceNoise": "noise0"
+  }
+ ],
+ "assetCondSha256": "a9ff66a4a910c9c1e94192f09d813af776f54014dd3eed4929247d14bd0372a8",
+ "sha256": "c67a9f7a52f2fcb992e44b7ebf53a5784a62e8f86e66b5708c2b0b1fcf68c44e",
+ "arrays": [
+  {
+   "name": "tpos",
+   "shape": [
+    51,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 0,
+   "byteLength": 2448
+  },
+  {
+   "name": "tposParents",
+   "shape": [
+    51,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 2560,
+   "byteLength": 2448
+  },
+  {
+   "name": "depths",
+   "shape": [
+    51
+   ],
+   "dtype": "int32",
+   "offset": 5120,
+   "byteLength": 204
+  },
+  {
+   "name": "jointNamesEmb",
+   "shape": [
+    51,
+    768
+   ],
+   "dtype": "float32",
+   "offset": 5376,
+   "byteLength": 156672
+  },
+  {
+   "name": "spectral",
+   "shape": [
+    51,
+    8
+   ],
+   "dtype": "float32",
+   "offset": 162048,
+   "byteLength": 1632
+  },
+  {
+   "name": "graphDist",
+   "shape": [
+    51,
+    51
+   ],
+   "dtype": "int32",
+   "offset": 163840,
+   "byteLength": 10404
+  },
+  {
+   "name": "jointRelations",
+   "shape": [
+    51,
+    51
+   ],
+   "dtype": "int32",
+   "offset": 174336,
+   "byteLength": 10404
+  },
+  {
+   "name": "mean",
+   "shape": [
+    51,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 184832,
+   "byteLength": 2448
+  },
+  {
+   "name": "std",
+   "shape": [
+    51,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 187392,
+   "byteLength": 2448
+  },
+  {
+   "name": "restPositions",
+   "shape": [
+    51,
+    3
+   ],
+   "dtype": "float32",
+   "offset": 189952,
+   "byteLength": 612
+  },
+  {
+   "name": "caption0",
+   "shape": [
+    768
+   ],
+   "dtype": "float32",
+   "offset": 190720,
+   "byteLength": 3072
+  },
+  {
+   "name": "noise0",
+   "shape": [
+    51,
+    12,
+    60
+   ],
+   "dtype": "float32",
+   "offset": 193792,
+   "byteLength": 146880
+  }
+ ],
+ "captureIdentity": {
+  "prompts": [
+   "An animal walks forward."
+  ],
+  "seed": 10,
+  "model_sha256": "f991ddd87262c251f4681d18297a6163ab4f139daa28c86d5ef0a66771df7bc6",
+  "config_sha256": "c8d3b44ee5ea6c5bd71e7b5ad1f4c94019314a71ac2578d59352618b6342374c",
+  "stats_sha256": "0ff2b9ef3ea9a15b024ab0670e55f659478251501e5596f9011d5bca4cf67773"
+ }
+}

skeletons/go2/demo.bin

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

skeletons/go2/demo.json

@@ -0,0 +1,292 @@
+{
+ "schema": "kaminos.unimate.demo-bundle.v0",
+ "skeleton": {
+  "name": "unitree_go2",
+  "joints": 18,
+  "parents": [
+   -1,
+   0,
+   0,
+   0,
+   0,
+   0,
+   1,
+   2,
+   3,
+   4,
+   6,
+   7,
+   8,
+   9,
+   10,
+   11,
+   12,
+   13
+  ],
+  "jointNames": [
+   "base",
+   "FR_hip",
+   "RL_hip",
+   "FL_hip",
+   "RR_hip",
+   "Head_upper",
+   "FR_thigh",
+   "RL_thigh",
+   "FL_thigh",
+   "RR_thigh",
+   "FR_calf",
+   "RL_calf",
+   "FL_calf",
+   "RR_calf",
+   "FR_foot",
+   "RL_foot",
+   "FL_foot",
+   "RR_foot"
+  ],
+  "fps": 30,
+  "frames": 60
+ },
+ "motionLength": 60,
+ "outputTimes": [
+  0.0,
+  0.020408162847161293,
+  0.040816325694322586,
+  0.06122449040412903,
+  0.08163265138864517,
+  0.10204081237316132,
+  0.12244898080825806,
+  0.1428571343421936,
+  0.16326530277729034,
+  0.18367347121238708,
+  0.20408162474632263,
+  0.22448979318141937,
+  0.2448979616165161,
+  0.26530611515045166,
+  0.2857142686843872,
+  0.30612245202064514,
+  0.3265306055545807,
+  0.34693875908851624,
+  0.36734694242477417,
+  0.3877550959587097,
+  0.40816324949264526,
+  0.4285714328289032,
+  0.44897958636283875,
+  0.4693877398967743,
+  0.4897959232330322,
+  0.5102040767669678,
+  0.5306122303009033,
+  0.5510204434394836,
+  0.5714285969734192,
+  0.5918367505073547,
+  0.6122449040412903,
+  0.6326530575752258,
+  0.6530612111091614,
+  0.6734694242477417,
+  0.6938775777816772,
+  0.7142857313156128,
+  0.7346938848495483,
+  0.7551020383834839,
+  0.7755101919174194,
+  0.795918345451355,
+  0.8163265585899353,
+  0.8367347121238708,
+  0.8571428656578064,
+  0.8775510191917419,
+  0.8979591727256775,
+  0.918367326259613,
+  0.9387755393981934,
+  0.9591836929321289,
+  0.9795918464660645,
+  1.0
+ ],
+ "prompts": [
+  {
+   "text": "An object trots forward.",
+   "caption": "caption0",
+   "referenceNoise": "noise0",
+   "referencePositions": "reference0"
+  },
+  {
+   "text": "An object rears up on its hind legs.",
+   "caption": "caption1",
+   "referenceNoise": "noise1",
+   "referencePositions": "reference1"
+  }
+ ],
+ "assetCondSha256": "46f56a7c5c73119327bc53d16b987307a22a839ad5e96f6a405e5ad0d5b9b4e4",
+ "sha256": "c3fc5b5aff8fb09d783dd946a5db8219dc5ecf72cfee453b79ffb54e62969bb0",
+ "arrays": [
+  {
+   "name": "tpos",
+   "shape": [
+    18,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 0,
+   "byteLength": 864
+  },
+  {
+   "name": "tposParents",
+   "shape": [
+    18,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 1024,
+   "byteLength": 864
+  },
+  {
+   "name": "depths",
+   "shape": [
+    18
+   ],
+   "dtype": "int32",
+   "offset": 2048,
+   "byteLength": 72
+  },
+  {
+   "name": "jointNamesEmb",
+   "shape": [
+    18,
+    768
+   ],
+   "dtype": "float32",
+   "offset": 2304,
+   "byteLength": 55296
+  },
+  {
+   "name": "spectral",
+   "shape": [
+    18,
+    8
+   ],
+   "dtype": "float32",
+   "offset": 57600,
+   "byteLength": 576
+  },
+  {
+   "name": "graphDist",
+   "shape": [
+    18,
+    18
+   ],
+   "dtype": "int32",
+   "offset": 58368,
+   "byteLength": 1296
+  },
+  {
+   "name": "jointRelations",
+   "shape": [
+    18,
+    18
+   ],
+   "dtype": "int32",
+   "offset": 59904,
+   "byteLength": 1296
+  },
+  {
+   "name": "mean",
+   "shape": [
+    18,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 61440,
+   "byteLength": 864
+  },
+  {
+   "name": "std",
+   "shape": [
+    18,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 62464,
+   "byteLength": 864
+  },
+  {
+   "name": "restPositions",
+   "shape": [
+    18,
+    3
+   ],
+   "dtype": "float32",
+   "offset": 63488,
+   "byteLength": 216
+  },
+  {
+   "name": "caption0",
+   "shape": [
+    768
+   ],
+   "dtype": "float32",
+   "offset": 63744,
+   "byteLength": 3072
+  },
+  {
+   "name": "noise0",
+   "shape": [
+    18,
+    12,
+    60
+   ],
+   "dtype": "float32",
+   "offset": 66816,
+   "byteLength": 51840
+  },
+  {
+   "name": "reference0",
+   "shape": [
+    60,
+    18,
+    3
+   ],
+   "dtype": "float32",
+   "offset": 118784,
+   "byteLength": 12960
+  },
+  {
+   "name": "caption1",
+   "shape": [
+    768
+   ],
+   "dtype": "float32",
+   "offset": 131840,
+   "byteLength": 3072
+  },
+  {
+   "name": "noise1",
+   "shape": [
+    18,
+    12,
+    60
+   ],
+   "dtype": "float32",
+   "offset": 134912,
+   "byteLength": 51840
+  },
+  {
+   "name": "reference1",
+   "shape": [
+    60,
+    18,
+    3
+   ],
+   "dtype": "float32",
+   "offset": 186880,
+   "byteLength": 12960
+  }
+ ],
+ "captureIdentity": {
+  "prompts": [
+   "An object trots forward.",
+   "An object rears up on its hind legs."
+  ],
+  "seed": 10,
+  "model_sha256": "f991ddd87262c251f4681d18297a6163ab4f139daa28c86d5ef0a66771df7bc6",
+  "config_sha256": "c8d3b44ee5ea6c5bd71e7b5ad1f4c94019314a71ac2578d59352618b6342374c",
+  "stats_sha256": "0ff2b9ef3ea9a15b024ab0670e55f659478251501e5596f9011d5bca4cf67773",
+  "asset_cond_sha256": "46f56a7c5c73119327bc53d16b987307a22a839ad5e96f6a405e5ad0d5b9b4e4"
+ }
+}

skeletons/horse/demo.bin

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

skeletons/horse/demo.json

@@ -0,0 +1,331 @@
+{
+ "schema": "kaminos.unimate.demo-bundle.v0",
+ "skeleton": {
+  "name": "Horse",
+  "joints": 63,
+  "parents": [
+   -1,
+   0,
+   0,
+   1,
+   1,
+   1,
+   2,
+   3,
+   3,
+   4,
+   5,
+   6,
+   7,
+   7,
+   7,
+   9,
+   10,
+   11,
+   12,
+   12,
+   12,
+   12,
+   13,
+   14,
+   15,
+   16,
+   17,
+   18,
+   18,
+   19,
+   20,
+   21,
+   22,
+   24,
+   25,
+   27,
+   27,
+   27,
+   27,
+   27,
+   27,
+   27,
+   28,
+   29,
+   30,
+   31,
+   35,
+   36,
+   42,
+   43,
+   44,
+   46,
+   49,
+   50,
+   51,
+   51,
+   51,
+   52,
+   53,
+   54,
+   54,
+   54,
+   59
+  ],
+  "jointNames": [
+   "Bip01_Pelvis",
+   "Bip01_Spine",
+   "BN_Tail_01",
+   "Bip01_Spine1",
+   "Bip01_R_Thigh",
+   "Bip01_L_Thigh",
+   "BN_Tail_02",
+   "Bip01_Spine2",
+   "BN_center",
+   "Bip01_R_Calf",
+   "Bip01_L_Calf",
+   "BN_Tail_03",
+   "Bip01_Neck",
+   "BN_hair04_01",
+   "Handle",
+   "Bip01_R_HorseLink",
+   "Bip01_L_HorseLink",
+   "BN_Tail_04",
+   "Bip01_Neck1",
+   "Bip01_L_Clavicle",
+   "Bip01_R_Clavicle",
+   "BN_hair03_01",
+   "BN_hair04_02",
+   "IK_Chain01",
+   "Bip01_R_Foot",
+   "Bip01_L_Foot",
+   "BN_Tail_05",
+   "Bip01_Head",
+   "BN_hair02_01",
+   "Bip01_L_UpperArm",
+   "Bip01_R_UpperArm",
+   "BN_hair03_02",
+   "BN_hair04_03",
+   "Bip01_R_Toe0",
+   "Bip01_L_Toe0",
+   "BN_Reins_01",
+   "BN_hair01_01",
+   "Bip01_Jaw",
+   "Bip01_R_Ear_01",
+   "Bip01__L_Ear_01",
+   "BN_Eyebrow_L",
+   "BN_Eyebrow_R",
+   "BN_hair02_02",
+   "Bip01_L_Forearm",
+   "Bip01_R_Forearm",
+   "BN_hair03_03",
+   "BN_Reins_02",
+   "BN_hair01_02",
+   "BN_hair02_03",
+   "Bip01_L_Hand",
+   "Bip01_R_Hand",
+   "BN_Reins_03",
+   "Bip01_L_Finger0",
+   "Bip01_R_Finger0",
+   "BN_Reins_04",
+   "BN_Halter_R_01",
+   "BN_Halter_L_01",
+   "Bip01_Xtra01",
+   "Bip01_Xtra02",
+   "BN_Reins_05",
+   "BN_Halter_R_02",
+   "BN_Halter_L_02",
+   "BN_Reins_06"
+  ],
+  "fps": 30,
+  "frames": 60
+ },
+ "motionLength": 60,
+ "outputTimes": [
+  0.0,
+  0.020408162847161293,
+  0.040816325694322586,
+  0.06122449040412903,
+  0.08163265138864517,
+  0.10204081237316132,
+  0.12244898080825806,
+  0.1428571343421936,
+  0.16326530277729034,
+  0.18367347121238708,
+  0.20408162474632263,
+  0.22448979318141937,
+  0.2448979616165161,
+  0.26530611515045166,
+  0.2857142686843872,
+  0.30612245202064514,
+  0.3265306055545807,
+  0.34693875908851624,
+  0.36734694242477417,
+  0.3877550959587097,
+  0.40816324949264526,
+  0.4285714328289032,
+  0.44897958636283875,
+  0.4693877398967743,
+  0.4897959232330322,
+  0.5102040767669678,
+  0.5306122303009033,
+  0.5510204434394836,
+  0.5714285969734192,
+  0.5918367505073547,
+  0.6122449040412903,
+  0.6326530575752258,
+  0.6530612111091614,
+  0.6734694242477417,
+  0.6938775777816772,
+  0.7142857313156128,
+  0.7346938848495483,
+  0.7551020383834839,
+  0.7755101919174194,
+  0.795918345451355,
+  0.8163265585899353,
+  0.8367347121238708,
+  0.8571428656578064,
+  0.8775510191917419,
+  0.8979591727256775,
+  0.918367326259613,
+  0.9387755393981934,
+  0.9591836929321289,
+  0.9795918464660645,
+  1.0
+ ],
+ "prompts": [
+  {
+   "text": "An animal walks forward.",
+   "caption": "caption0",
+   "referenceNoise": "noise0"
+  }
+ ],
+ "assetCondSha256": "a9ff66a4a910c9c1e94192f09d813af776f54014dd3eed4929247d14bd0372a8",
+ "sha256": "bc2cf461ee92eaa7c8bebc2bceafe926af66012c0ca83195234d584dda0b13a9",
+ "arrays": [
+  {
+   "name": "tpos",
+   "shape": [
+    63,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 0,
+   "byteLength": 3024
+  },
+  {
+   "name": "tposParents",
+   "shape": [
+    63,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 3072,
+   "byteLength": 3024
+  },
+  {
+   "name": "depths",
+   "shape": [
+    63
+   ],
+   "dtype": "int32",
+   "offset": 6144,
+   "byteLength": 252
+  },
+  {
+   "name": "jointNamesEmb",
+   "shape": [
+    63,
+    768
+   ],
+   "dtype": "float32",
+   "offset": 6400,
+   "byteLength": 193536
+  },
+  {
+   "name": "spectral",
+   "shape": [
+    63,
+    8
+   ],
+   "dtype": "float32",
+   "offset": 199936,
+   "byteLength": 2016
+  },
+  {
+   "name": "graphDist",
+   "shape": [
+    63,
+    63
+   ],
+   "dtype": "int32",
+   "offset": 201984,
+   "byteLength": 15876
+  },
+  {
+   "name": "jointRelations",
+   "shape": [
+    63,
+    63
+   ],
+   "dtype": "int32",
+   "offset": 218112,
+   "byteLength": 15876
+  },
+  {
+   "name": "mean",
+   "shape": [
+    63,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 234240,
+   "byteLength": 3024
+  },
+  {
+   "name": "std",
+   "shape": [
+    63,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 237312,
+   "byteLength": 3024
+  },
+  {
+   "name": "restPositions",
+   "shape": [
+    63,
+    3
+   ],
+   "dtype": "float32",
+   "offset": 240384,
+   "byteLength": 756
+  },
+  {
+   "name": "caption0",
+   "shape": [
+    768
+   ],
+   "dtype": "float32",
+   "offset": 241152,
+   "byteLength": 3072
+  },
+  {
+   "name": "noise0",
+   "shape": [
+    63,
+    12,
+    60
+   ],
+   "dtype": "float32",
+   "offset": 244224,
+   "byteLength": 181440
+  }
+ ],
+ "captureIdentity": {
+  "prompts": [
+   "An animal walks forward."
+  ],
+  "seed": 10,
+  "model_sha256": "f991ddd87262c251f4681d18297a6163ab4f139daa28c86d5ef0a66771df7bc6",
+  "config_sha256": "c8d3b44ee5ea6c5bd71e7b5ad1f4c94019314a71ac2578d59352618b6342374c",
+  "stats_sha256": "0ff2b9ef3ea9a15b024ab0670e55f659478251501e5596f9011d5bca4cf67773"
+ }
+}

skeletons/index.json

@@ -0,0 +1,83 @@
+{
+ "schema": "kaminos.unimate.skeleton-index.v0",
+ "skeletons": [
+  {
+   "key": "go2",
+   "name": "unitree_go2",
+   "joints": 18,
+   "label": "Go2 robot dog",
+   "suggestions": [
+    "An object trots forward.",
+    "An object rears up on its hind legs.",
+    "An object jumps forward."
+   ]
+  },
+  {
+   "key": "cat",
+   "name": "Cat",
+   "joints": 43,
+   "label": "Cat",
+   "suggestions": [
+    "A cat walks forward.",
+    "A cat jumps forward."
+   ]
+  },
+  {
+   "key": "dog",
+   "name": "Dog",
+   "joints": 51,
+   "label": "Dog",
+   "suggestions": [
+    "A dog runs forward.",
+    "A dog jumps forward."
+   ]
+  },
+  {
+   "key": "horse",
+   "name": "Horse",
+   "joints": 63,
+   "label": "Horse",
+   "suggestions": [
+    "A horse rears up on its hind legs.",
+    "A horse gallops forward."
+   ]
+  },
+  {
+   "key": "trex",
+   "name": "Trex",
+   "joints": 60,
+   "label": "T-rex",
+   "suggestions": [
+    "A tyrannosaurus roars.",
+    "A tyrannosaurus walks forward."
+   ]
+  },
+  {
+   "key": "spider",
+   "name": "Spider",
+   "joints": 69,
+   "label": "Spider",
+   "suggestions": [
+    "A spider crawls forward."
+   ]
+  },
+  {
+   "key": "eagle",
+   "name": "Eagle",
+   "joints": 51,
+   "label": "Eagle",
+   "suggestions": [
+    "An eagle flaps its wings."
+   ]
+  },
+  {
+   "key": "parrot",
+   "name": "Parrot",
+   "joints": 70,
+   "label": "Parrot",
+   "suggestions": [
+    "A parrot flaps its wings."
+   ]
+  }
+ ]
+}

skeletons/parrot/demo.bin

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

skeletons/parrot/demo.json

@@ -0,0 +1,345 @@
+{
+ "schema": "kaminos.unimate.demo-bundle.v0",
+ "skeleton": {
+  "name": "Parrot",
+  "joints": 70,
+  "parents": [
+   -1,
+   0,
+   0,
+   1,
+   1,
+   1,
+   2,
+   3,
+   3,
+   3,
+   4,
+   5,
+   6,
+   6,
+   6,
+   7,
+   8,
+   8,
+   9,
+   9,
+   10,
+   11,
+   12,
+   15,
+   16,
+   16,
+   18,
+   18,
+   20,
+   21,
+   22,
+   23,
+   23,
+   23,
+   23,
+   23,
+   23,
+   23,
+   23,
+   23,
+   24,
+   25,
+   26,
+   27,
+   28,
+   28,
+   28,
+   28,
+   29,
+   29,
+   29,
+   29,
+   31,
+   32,
+   44,
+   45,
+   46,
+   47,
+   48,
+   49,
+   50,
+   51,
+   54,
+   55,
+   56,
+   57,
+   58,
+   59,
+   60,
+   61
+  ],
+  "jointNames": [
+   "Bip01_Pelvis",
+   "Bip01_Spine",
+   "BN_Tail_01",
+   "Bip01_Spine1",
+   "Bip01_L_Thigh",
+   "Bip01_R_Thigh",
+   "BN_Tail_02",
+   "Bip01_Neck",
+   "BN_Wing_L_01",
+   "BN_Wing_R_01",
+   "Bip01_L_Calf",
+   "Bip01_R_Calf",
+   "BN_Tail_03",
+   "BN_Tail_06",
+   "BN_Tail_07",
+   "Bip01_Neck1",
+   "BN_Wing_L_02",
+   "BN_Wing_L_05",
+   "BN_Wing_R_02",
+   "BN_Wing_R_05",
+   "Bip01_L_HorseLink",
+   "Bip01_R_HorseLink",
+   "BN_Tail_04",
+   "Bip01_Head",
+   "BN_Wing_L_06",
+   "BN_Wing_L_03",
+   "BN_Wing_R_03",
+   "BN_Wing_R_06",
+   "Bip01_L_Foot",
+   "Bip01_R_Foot",
+   "BN_Tail_05",
+   "BN_Fur_03",
+   "BN_Fur_01",
+   "BN_Jaw",
+   "BN_Eye_L_01",
+   "BN_Eye_L_02",
+   "BN_Eye_R_02",
+   "BN_Eye_R_01",
+   "BN_Fur_05",
+   "Bip01_HeadNub",
+   "BN_Wing_L_07",
+   "BN_Wing_L_04",
+   "BN_Wing_R_04",
+   "BN_Wing_R_07",
+   "Bip01_L_Toe3",
+   "Bip01_L_Toe2",
+   "Bip01_L_Toe1",
+   "Bip01_L_Toe0",
+   "Bip01_R_Toe3",
+   "Bip01_R_Toe2",
+   "Bip01_R_Toe1",
+   "Bip01_R_Toe0",
+   "BN_Fur_04",
+   "BN_Fur_02",
+   "Bip01_L_Toe31",
+   "Bip01_L_Toe21",
+   "Bip01_L_Toe11",
+   "Bip01_L_Toe01",
+   "Bip01_R_Toe31",
+   "Bip01_R_Toe21",
+   "Bip01_R_Toe11",
+   "Bip01_R_Toe01",
+   "Bip01_L_Toe3Nub",
+   "Bip01_L_Toe2Nub",
+   "Bip01_L_Toe1Nub",
+   "Bip01_L_Toe0Nub",
+   "Bip01_R_Toe3Nub",
+   "Bip01_R_Toe2Nub",
+   "Bip01_R_Toe1Nub",
+   "Bip01_R_Toe0Nub"
+  ],
+  "fps": 30,
+  "frames": 60
+ },
+ "motionLength": 60,
+ "outputTimes": [
+  0.0,
+  0.020408162847161293,
+  0.040816325694322586,
+  0.06122449040412903,
+  0.08163265138864517,
+  0.10204081237316132,
+  0.12244898080825806,
+  0.1428571343421936,
+  0.16326530277729034,
+  0.18367347121238708,
+  0.20408162474632263,
+  0.22448979318141937,
+  0.2448979616165161,
+  0.26530611515045166,
+  0.2857142686843872,
+  0.30612245202064514,
+  0.3265306055545807,
+  0.34693875908851624,
+  0.36734694242477417,
+  0.3877550959587097,
+  0.40816324949264526,
+  0.4285714328289032,
+  0.44897958636283875,
+  0.4693877398967743,
+  0.4897959232330322,
+  0.5102040767669678,
+  0.5306122303009033,
+  0.5510204434394836,
+  0.5714285969734192,
+  0.5918367505073547,
+  0.6122449040412903,
+  0.6326530575752258,
+  0.6530612111091614,
+  0.6734694242477417,
+  0.6938775777816772,
+  0.7142857313156128,
+  0.7346938848495483,
+  0.7551020383834839,
+  0.7755101919174194,
+  0.795918345451355,
+  0.8163265585899353,
+  0.8367347121238708,
+  0.8571428656578064,
+  0.8775510191917419,
+  0.8979591727256775,
+  0.918367326259613,
+  0.9387755393981934,
+  0.9591836929321289,
+  0.9795918464660645,
+  1.0
+ ],
+ "prompts": [
+  {
+   "text": "A parrot flaps its wings.",
+   "caption": "caption0",
+   "referenceNoise": "noise0"
+  }
+ ],
+ "assetCondSha256": "a9ff66a4a910c9c1e94192f09d813af776f54014dd3eed4929247d14bd0372a8",
+ "sha256": "cb9247bc69292e8799551f9470821f789f372a8744bf391e87f0e942ee5110d8",
+ "arrays": [
+  {
+   "name": "tpos",
+   "shape": [
+    70,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 0,
+   "byteLength": 3360
+  },
+  {
+   "name": "tposParents",
+   "shape": [
+    70,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 3584,
+   "byteLength": 3360
+  },
+  {
+   "name": "depths",
+   "shape": [
+    70
+   ],
+   "dtype": "int32",
+   "offset": 7168,
+   "byteLength": 280
+  },
+  {
+   "name": "jointNamesEmb",
+   "shape": [
+    70,
+    768
+   ],
+   "dtype": "float32",
+   "offset": 7680,
+   "byteLength": 215040
+  },
+  {
+   "name": "spectral",
+   "shape": [
+    70,
+    8
+   ],
+   "dtype": "float32",
+   "offset": 222720,
+   "byteLength": 2240
+  },
+  {
+   "name": "graphDist",
+   "shape": [
+    70,
+    70
+   ],
+   "dtype": "int32",
+   "offset": 225024,
+   "byteLength": 19600
+  },
+  {
+   "name": "jointRelations",
+   "shape": [
+    70,
+    70
+   ],
+   "dtype": "int32",
+   "offset": 244736,
+   "byteLength": 19600
+  },
+  {
+   "name": "mean",
+   "shape": [
+    70,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 264448,
+   "byteLength": 3360
+  },
+  {
+   "name": "std",
+   "shape": [
+    70,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 268032,
+   "byteLength": 3360
+  },
+  {
+   "name": "restPositions",
+   "shape": [
+    70,
+    3
+   ],
+   "dtype": "float32",
+   "offset": 271616,
+   "byteLength": 840
+  },
+  {
+   "name": "caption0",
+   "shape": [
+    768
+   ],
+   "dtype": "float32",
+   "offset": 272640,
+   "byteLength": 3072
+  },
+  {
+   "name": "noise0",
+   "shape": [
+    70,
+    12,
+    60
+   ],
+   "dtype": "float32",
+   "offset": 275712,
+   "byteLength": 201600
+  }
+ ],
+ "captureIdentity": {
+  "prompts": [
+   "A parrot flaps its wings."
+  ],
+  "seed": 10,
+  "model_sha256": "f991ddd87262c251f4681d18297a6163ab4f139daa28c86d5ef0a66771df7bc6",
+  "config_sha256": "c8d3b44ee5ea6c5bd71e7b5ad1f4c94019314a71ac2578d59352618b6342374c",
+  "stats_sha256": "0ff2b9ef3ea9a15b024ab0670e55f659478251501e5596f9011d5bca4cf67773"
+ }
+}

skeletons/spider/demo.bin

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

skeletons/spider/demo.json

@@ -0,0 +1,343 @@
+{
+ "schema": "kaminos.unimate.demo-bundle.v0",
+ "skeleton": {
+  "name": "Spider",
+  "joints": 69,
+  "parents": [
+   -1,
+   0,
+   0,
+   0,
+   0,
+   0,
+   0,
+   0,
+   0,
+   0,
+   0,
+   0,
+   0,
+   0,
+   0,
+   0,
+   0,
+   1,
+   2,
+   3,
+   4,
+   5,
+   6,
+   7,
+   8,
+   9,
+   10,
+   11,
+   12,
+   13,
+   17,
+   18,
+   19,
+   20,
+   21,
+   22,
+   23,
+   24,
+   25,
+   26,
+   27,
+   28,
+   29,
+   30,
+   31,
+   32,
+   33,
+   34,
+   35,
+   36,
+   37,
+   38,
+   39,
+   43,
+   44,
+   45,
+   46,
+   47,
+   48,
+   49,
+   50,
+   53,
+   54,
+   55,
+   56,
+   57,
+   58,
+   59,
+   60
+  ],
+  "jointNames": [
+   "_body_",
+   "Leg_L_10_",
+   "Leg_R_10_",
+   "Leg_L_20_",
+   "Leg_R_20_",
+   "Leg_L_00_",
+   "Leg_R_00_",
+   "Leg_R_30_",
+   "Leg_L_30_",
+   "ArmRCollarbone",
+   "ArmLCollarbone",
+   "Tail1",
+   "FangR_00_",
+   "FangL_00_",
+   "R_Jaw_",
+   "L_Jaw_",
+   "NPC_L_MagicNode__LMag_",
+   "Leg_L_11_",
+   "Leg_R_11_",
+   "Leg_L_21_",
+   "Leg_R_21_",
+   "Leg_L_01_",
+   "Leg_R_01_",
+   "Leg_R_31_",
+   "Leg_L_31_",
+   "ArmR_01_",
+   "ArmL_01_",
+   "Tail2",
+   "FangR_01_",
+   "FangL_01_",
+   "Leg_L_12_",
+   "Leg_R_12_",
+   "Leg_L_22_",
+   "Leg_R_22_",
+   "Leg_L_02_",
+   "Leg_R_02_",
+   "Leg_R_32_",
+   "Leg_L_32_",
+   "ArmR_02_",
+   "ArmL_02_",
+   "Tail3",
+   "FangR_02_",
+   "FangL_02_",
+   "Leg_L_13_",
+   "Leg_R_13_",
+   "Leg_L_23_",
+   "Leg_R_23_",
+   "Leg_L_03_",
+   "Leg_R_03_",
+   "Leg_R_33_",
+   "Leg_L_33_",
+   "ArmRClaw",
+   "ArmLClaw",
+   "Leg_L_14_",
+   "Leg_R_14_",
+   "Leg_L_24_",
+   "Leg_R_24_",
+   "Leg_L_04_",
+   "Leg_R_04_",
+   "Leg_R_34_",
+   "Leg_L_34_",
+   "_LToe1_",
+   "_RToe1_",
+   "_LToe2_",
+   "_RToe2_",
+   "_LToe0_",
+   "_RToe0_",
+   "_RToe3_",
+   "_LToe3_"
+  ],
+  "fps": 30,
+  "frames": 60
+ },
+ "motionLength": 60,
+ "outputTimes": [
+  0.0,
+  0.020408162847161293,
+  0.040816325694322586,
+  0.06122449040412903,
+  0.08163265138864517,
+  0.10204081237316132,
+  0.12244898080825806,
+  0.1428571343421936,
+  0.16326530277729034,
+  0.18367347121238708,
+  0.20408162474632263,
+  0.22448979318141937,
+  0.2448979616165161,
+  0.26530611515045166,
+  0.2857142686843872,
+  0.30612245202064514,
+  0.3265306055545807,
+  0.34693875908851624,
+  0.36734694242477417,
+  0.3877550959587097,
+  0.40816324949264526,
+  0.4285714328289032,
+  0.44897958636283875,
+  0.4693877398967743,
+  0.4897959232330322,
+  0.5102040767669678,
+  0.5306122303009033,
+  0.5510204434394836,
+  0.5714285969734192,
+  0.5918367505073547,
+  0.6122449040412903,
+  0.6326530575752258,
+  0.6530612111091614,
+  0.6734694242477417,
+  0.6938775777816772,
+  0.7142857313156128,
+  0.7346938848495483,
+  0.7551020383834839,
+  0.7755101919174194,
+  0.795918345451355,
+  0.8163265585899353,
+  0.8367347121238708,
+  0.8571428656578064,
+  0.8775510191917419,
+  0.8979591727256775,
+  0.918367326259613,
+  0.9387755393981934,
+  0.9591836929321289,
+  0.9795918464660645,
+  1.0
+ ],
+ "prompts": [
+  {
+   "text": "An animal walks forward.",
+   "caption": "caption0",
+   "referenceNoise": "noise0"
+  }
+ ],
+ "assetCondSha256": "a9ff66a4a910c9c1e94192f09d813af776f54014dd3eed4929247d14bd0372a8",
+ "sha256": "a1347949704a738c700e8065259917e8dc308d9232ac9b7c713b3657bbbd39b7",
+ "arrays": [
+  {
+   "name": "tpos",
+   "shape": [
+    69,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 0,
+   "byteLength": 3312
+  },
+  {
+   "name": "tposParents",
+   "shape": [
+    69,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 3328,
+   "byteLength": 3312
+  },
+  {
+   "name": "depths",
+   "shape": [
+    69
+   ],
+   "dtype": "int32",
+   "offset": 6656,
+   "byteLength": 276
+  },
+  {
+   "name": "jointNamesEmb",
+   "shape": [
+    69,
+    768
+   ],
+   "dtype": "float32",
+   "offset": 7168,
+   "byteLength": 211968
+  },
+  {
+   "name": "spectral",
+   "shape": [
+    69,
+    8
+   ],
+   "dtype": "float32",
+   "offset": 219136,
+   "byteLength": 2208
+  },
+  {
+   "name": "graphDist",
+   "shape": [
+    69,
+    69
+   ],
+   "dtype": "int32",
+   "offset": 221440,
+   "byteLength": 19044
+  },
+  {
+   "name": "jointRelations",
+   "shape": [
+    69,
+    69
+   ],
+   "dtype": "int32",
+   "offset": 240640,
+   "byteLength": 19044
+  },
+  {
+   "name": "mean",
+   "shape": [
+    69,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 259840,
+   "byteLength": 3312
+  },
+  {
+   "name": "std",
+   "shape": [
+    69,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 263168,
+   "byteLength": 3312
+  },
+  {
+   "name": "restPositions",
+   "shape": [
+    69,
+    3
+   ],
+   "dtype": "float32",
+   "offset": 266496,
+   "byteLength": 828
+  },
+  {
+   "name": "caption0",
+   "shape": [
+    768
+   ],
+   "dtype": "float32",
+   "offset": 267520,
+   "byteLength": 3072
+  },
+  {
+   "name": "noise0",
+   "shape": [
+    69,
+    12,
+    60
+   ],
+   "dtype": "float32",
+   "offset": 270592,
+   "byteLength": 198720
+  }
+ ],
+ "captureIdentity": {
+  "prompts": [
+   "An animal walks forward."
+  ],
+  "seed": 10,
+  "model_sha256": "f991ddd87262c251f4681d18297a6163ab4f139daa28c86d5ef0a66771df7bc6",
+  "config_sha256": "c8d3b44ee5ea6c5bd71e7b5ad1f4c94019314a71ac2578d59352618b6342374c",
+  "stats_sha256": "0ff2b9ef3ea9a15b024ab0670e55f659478251501e5596f9011d5bca4cf67773"
+ }
+}

skeletons/trex/demo.bin

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

skeletons/trex/demo.json

@@ -0,0 +1,325 @@
+{
+ "schema": "kaminos.unimate.demo-bundle.v0",
+ "skeleton": {
+  "name": "Trex",
+  "joints": 60,
+  "parents": [
+   -1,
+   0,
+   0,
+   1,
+   1,
+   1,
+   2,
+   3,
+   4,
+   5,
+   6,
+   6,
+   6,
+   7,
+   7,
+   8,
+   8,
+   9,
+   10,
+   11,
+   12,
+   13,
+   15,
+   17,
+   18,
+   18,
+   19,
+   20,
+   21,
+   22,
+   23,
+   24,
+   26,
+   27,
+   28,
+   28,
+   28,
+   29,
+   29,
+   29,
+   30,
+   31,
+   31,
+   31,
+   32,
+   32,
+   33,
+   33,
+   34,
+   35,
+   36,
+   37,
+   38,
+   39,
+   41,
+   44,
+   45,
+   46,
+   47,
+   54
+  ],
+  "jointNames": [
+   "jt_Cog_C",
+   "jt_Hips_C",
+   "jt_Spine1_C",
+   "jt_Hip_R",
+   "jt_Hip_L",
+   "jt_Tail1_C",
+   "jt_Spine2_C",
+   "jt_Thigh_R",
+   "jt_Thigh_L",
+   "jt_Tail2_C",
+   "jt_Neck1_C",
+   "jt_Clavicle_R",
+   "jt_Clavicle_L",
+   "jt_Knee_R",
+   "jt_ThighMuscle_R",
+   "jt_Knee_L",
+   "jt_ThighMuscle_L",
+   "jt_Tail3_C",
+   "jt_Neck2_C",
+   "jt_Shoulder_R",
+   "jt_Shoulder_L",
+   "jt_Ankle_R",
+   "jt_Ankle_L",
+   "jt_Tail4_C",
+   "jt_Neck3_C",
+   "jt_NeckMuscle_C",
+   "jt_Elbow_R",
+   "jt_Elbow_L",
+   "jt_Foot_R",
+   "jt_Foot_L",
+   "jt_Tail5_C",
+   "jt_Head_C",
+   "jt_Wrist_R",
+   "jt_Wrist_L",
+   "jt_ToeMiddle_R",
+   "jt_ToeInner_R",
+   "jt_ToeOutter_R",
+   "jt_ToeMiddle_L",
+   "jt_ToeInner_L",
+   "jt_ToeOutter_L",
+   "jt_Tail6_C",
+   "jt_Jaw_C",
+   "jt_Eye_L",
+   "jt_Eye_R",
+   "jt_FingerInner_R",
+   "jt_FingerOutter_R",
+   "jt_FingerInner_L",
+   "jt_FingerOutter_L",
+   "jt_ClawMiddle_R",
+   "jt_ClawInner_R",
+   "jt_ClawOutter_R",
+   "jt_ClawMiddle_L",
+   "jt_ClawInner_L",
+   "jt_ClawOutter_L",
+   "jt_Tongue1_C",
+   "jt_HandClawInner_R",
+   "jt_HandClawOutter_R",
+   "jt_HandClawInner_L",
+   "jt_HandClawOutter_L",
+   "jt_Tongue2_C"
+  ],
+  "fps": 30,
+  "frames": 60
+ },
+ "motionLength": 60,
+ "outputTimes": [
+  0.0,
+  0.020408162847161293,
+  0.040816325694322586,
+  0.06122449040412903,
+  0.08163265138864517,
+  0.10204081237316132,
+  0.12244898080825806,
+  0.1428571343421936,
+  0.16326530277729034,
+  0.18367347121238708,
+  0.20408162474632263,
+  0.22448979318141937,
+  0.2448979616165161,
+  0.26530611515045166,
+  0.2857142686843872,
+  0.30612245202064514,
+  0.3265306055545807,
+  0.34693875908851624,
+  0.36734694242477417,
+  0.3877550959587097,
+  0.40816324949264526,
+  0.4285714328289032,
+  0.44897958636283875,
+  0.4693877398967743,
+  0.4897959232330322,
+  0.5102040767669678,
+  0.5306122303009033,
+  0.5510204434394836,
+  0.5714285969734192,
+  0.5918367505073547,
+  0.6122449040412903,
+  0.6326530575752258,
+  0.6530612111091614,
+  0.6734694242477417,
+  0.6938775777816772,
+  0.7142857313156128,
+  0.7346938848495483,
+  0.7551020383834839,
+  0.7755101919174194,
+  0.795918345451355,
+  0.8163265585899353,
+  0.8367347121238708,
+  0.8571428656578064,
+  0.8775510191917419,
+  0.8979591727256775,
+  0.918367326259613,
+  0.9387755393981934,
+  0.9591836929321289,
+  0.9795918464660645,
+  1.0
+ ],
+ "prompts": [
+  {
+   "text": "An animal walks forward.",
+   "caption": "caption0",
+   "referenceNoise": "noise0"
+  }
+ ],
+ "assetCondSha256": "a9ff66a4a910c9c1e94192f09d813af776f54014dd3eed4929247d14bd0372a8",
+ "sha256": "3edcb95f0eb3ffcca4187cda2b4d6167bc386907693c5c0449ecb55db0c553e7",
+ "arrays": [
+  {
+   "name": "tpos",
+   "shape": [
+    60,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 0,
+   "byteLength": 2880
+  },
+  {
+   "name": "tposParents",
+   "shape": [
+    60,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 3072,
+   "byteLength": 2880
+  },
+  {
+   "name": "depths",
+   "shape": [
+    60
+   ],
+   "dtype": "int32",
+   "offset": 6144,
+   "byteLength": 240
+  },
+  {
+   "name": "jointNamesEmb",
+   "shape": [
+    60,
+    768
+   ],
+   "dtype": "float32",
+   "offset": 6400,
+   "byteLength": 184320
+  },
+  {
+   "name": "spectral",
+   "shape": [
+    60,
+    8
+   ],
+   "dtype": "float32",
+   "offset": 190720,
+   "byteLength": 1920
+  },
+  {
+   "name": "graphDist",
+   "shape": [
+    60,
+    60
+   ],
+   "dtype": "int32",
+   "offset": 192768,
+   "byteLength": 14400
+  },
+  {
+   "name": "jointRelations",
+   "shape": [
+    60,
+    60
+   ],
+   "dtype": "int32",
+   "offset": 207360,
+   "byteLength": 14400
+  },
+  {
+   "name": "mean",
+   "shape": [
+    60,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 221952,
+   "byteLength": 2880
+  },
+  {
+   "name": "std",
+   "shape": [
+    60,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 225024,
+   "byteLength": 2880
+  },
+  {
+   "name": "restPositions",
+   "shape": [
+    60,
+    3
+   ],
+   "dtype": "float32",
+   "offset": 228096,
+   "byteLength": 720
+  },
+  {
+   "name": "caption0",
+   "shape": [
+    768
+   ],
+   "dtype": "float32",
+   "offset": 228864,
+   "byteLength": 3072
+  },
+  {
+   "name": "noise0",
+   "shape": [
+    60,
+    12,
+    60
+   ],
+   "dtype": "float32",
+   "offset": 231936,
+   "byteLength": 172800
+  }
+ ],
+ "captureIdentity": {
+  "prompts": [
+   "An animal walks forward."
+  ],
+  "seed": 10,
+  "model_sha256": "f991ddd87262c251f4681d18297a6163ab4f139daa28c86d5ef0a66771df7bc6",
+  "config_sha256": "c8d3b44ee5ea6c5bd71e7b5ad1f4c94019314a71ac2578d59352618b6342374c",
+  "stats_sha256": "0ff2b9ef3ea9a15b024ab0670e55f659478251501e5596f9011d5bca4cf67773"
+ }
+}

t5/t5-encoder-fp32.bin

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

t5/t5-manifest.json

@@ -0,0 +1,1703 @@
+{
+ "schema": "kaminos.unimate.t5-encoder-weights.v0",
+ "file": "t5-encoder-fp32.bin",
+ "byteLength": 438514176,
+ "sha256": "91b8c5526ccdfdebb17cdf700ed0d454c680f4e79a4cea88a62728a4bdaaf221",
+ "config": {
+  "d_model": 768,
+  "d_kv": 64,
+  "d_ff": 2048,
+  "num_heads": 12,
+  "num_layers": 12,
+  "relative_attention_num_buckets": 32,
+  "relative_attention_max_distance": 128,
+  "layer_norm_epsilon": 1e-06,
+  "feed_forward_proj": "gated-gelu",
+  "vocab_size": 32128
+ },
+ "source": {
+  "model_sha256": "1dfb70afdcedceb9f9fae2f9b68e004ad934361fb35b9b2bd50b45ea90790fc8",
+  "model": "google/flan-t5-base",
+  "license": "Apache-2.0"
+ },
+ "tensors": [
+  {
+   "name": "shared",
+   "shape": [
+    32128,
+    768
+   ],
+   "offset": 0,
+   "byteLength": 98697216
+  },
+  {
+   "name": "relative_attention_bias",
+   "shape": [
+    32,
+    12
+   ],
+   "offset": 98697216,
+   "byteLength": 1536
+  },
+  {
+   "name": "final_layer_norm",
+   "shape": [
+    768
+   ],
+   "offset": 98698752,
+   "byteLength": 3072
+  },
+  {
+   "name": "block0.ln_attn",
+   "shape": [
+    768
+   ],
+   "offset": 98701824,
+   "byteLength": 3072
+  },
+  {
+   "name": "block0.qkv",
+   "shape": [
+    2304,
+    768
+   ],
+   "offset": 98704896,
+   "byteLength": 7077888
+  },
+  {
+   "name": "block0.o",
+   "shape": [
+    768,
+    768
+   ],
+   "offset": 105782784,
+   "byteLength": 2359296
+  },
+  {
+   "name": "block0.ln_ff",
+   "shape": [
+    768
+   ],
+   "offset": 108142080,
+   "byteLength": 3072
+  },
+  {
+   "name": "block0.wi",
+   "shape": [
+    4096,
+    768
+   ],
+   "offset": 108145152,
+   "byteLength": 12582912
+  },
+  {
+   "name": "block0.wo",
+   "shape": [
+    768,
+    2048
+   ],
+   "offset": 120728064,
+   "byteLength": 6291456
+  },
+  {
+   "name": "block1.ln_attn",
+   "shape": [
+    768
+   ],
+   "offset": 127019520,
+   "byteLength": 3072
+  },
+  {
+   "name": "block1.qkv",
+   "shape": [
+    2304,
+    768
+   ],
+   "offset": 127022592,
+   "byteLength": 7077888
+  },
+  {
+   "name": "block1.o",
+   "shape": [
+    768,
+    768
+   ],
+   "offset": 134100480,
+   "byteLength": 2359296
+  },
+  {
+   "name": "block1.ln_ff",
+   "shape": [
+    768
+   ],
+   "offset": 136459776,
+   "byteLength": 3072
+  },
+  {
+   "name": "block1.wi",
+   "shape": [
+    4096,
+    768
+   ],
+   "offset": 136462848,
+   "byteLength": 12582912
+  },
+  {
+   "name": "block1.wo",
+   "shape": [
+    768,
+    2048
+   ],
+   "offset": 149045760,
+   "byteLength": 6291456
+  },
+  {
+   "name": "block2.ln_attn",
+   "shape": [
+    768
+   ],
+   "offset": 155337216,
+   "byteLength": 3072
+  },
+  {
+   "name": "block2.qkv",
+   "shape": [
+    2304,
+    768
+   ],
+   "offset": 155340288,
+   "byteLength": 7077888
+  },
+  {
+   "name": "block2.o",
+   "shape": [
+    768,
+    768
+   ],
+   "offset": 162418176,
+   "byteLength": 2359296
+  },
+  {
+   "name": "block2.ln_ff",
+   "shape": [
+    768
+   ],
+   "offset": 164777472,
+   "byteLength": 3072
+  },
+  {
+   "name": "block2.wi",
+   "shape": [
+    4096,
+    768
+   ],
+   "offset": 164780544,
+   "byteLength": 12582912
+  },
+  {
+   "name": "block2.wo",
+   "shape": [
+    768,
+    2048
+   ],
+   "offset": 177363456,
+   "byteLength": 6291456
+  },
+  {
+   "name": "block3.ln_attn",
+   "shape": [
+    768
+   ],
+   "offset": 183654912,
+   "byteLength": 3072
+  },
+  {
+   "name": "block3.qkv",
+   "shape": [
+    2304,
+    768
+   ],
+   "offset": 183657984,
+   "byteLength": 7077888
+  },
+  {
+   "name": "block3.o",
+   "shape": [
+    768,
+    768
+   ],
+   "offset": 190735872,
+   "byteLength": 2359296
+  },
+  {
+   "name": "block3.ln_ff",
+   "shape": [
+    768
+   ],
+   "offset": 193095168,
+   "byteLength": 3072
+  },
+  {
+   "name": "block3.wi",
+   "shape": [
+    4096,
+    768
+   ],
+   "offset": 193098240,
+   "byteLength": 12582912
+  },
+  {
+   "name": "block3.wo",
+   "shape": [
+    768,
+    2048
+   ],
+   "offset": 205681152,
+   "byteLength": 6291456
+  },
+  {
+   "name": "block4.ln_attn",
+   "shape": [
+    768
+   ],
+   "offset": 211972608,
+   "byteLength": 3072
+  },
+  {
+   "name": "block4.qkv",
+   "shape": [
+    2304,
+    768
+   ],
+   "offset": 211975680,
+   "byteLength": 7077888
+  },
+  {
+   "name": "block4.o",
+   "shape": [
+    768,
+    768
+   ],
+   "offset": 219053568,
+   "byteLength": 2359296
+  },
+  {
+   "name": "block4.ln_ff",
+   "shape": [
+    768
+   ],
+   "offset": 221412864,
+   "byteLength": 3072
+  },
+  {
+   "name": "block4.wi",
+   "shape": [
+    4096,
+    768
+   ],
+   "offset": 221415936,
+   "byteLength": 12582912
+  },
+  {
+   "name": "block4.wo",
+   "shape": [
+    768,
+    2048
+   ],
+   "offset": 233998848,
+   "byteLength": 6291456
+  },
+  {
+   "name": "block5.ln_attn",
+   "shape": [
+    768
+   ],
+   "offset": 240290304,
+   "byteLength": 3072
+  },
+  {
+   "name": "block5.qkv",
+   "shape": [
+    2304,
+    768
+   ],
+   "offset": 240293376,
+   "byteLength": 7077888
+  },
+  {
+   "name": "block5.o",
+   "shape": [
+    768,
+    768
+   ],
+   "offset": 247371264,
+   "byteLength": 2359296
+  },
+  {
+   "name": "block5.ln_ff",
+   "shape": [
+    768
+   ],
+   "offset": 249730560,
+   "byteLength": 3072
+  },
+  {
+   "name": "block5.wi",
+   "shape": [
+    4096,
+    768
+   ],
+   "offset": 249733632,
+   "byteLength": 12582912
+  },
+  {
+   "name": "block5.wo",
+   "shape": [
+    768,
+    2048
+   ],
+   "offset": 262316544,
+   "byteLength": 6291456
+  },
+  {
+   "name": "block6.ln_attn",
+   "shape": [
+    768
+   ],
+   "offset": 268608000,
+   "byteLength": 3072
+  },
+  {
+   "name": "block6.qkv",
+   "shape": [
+    2304,
+    768
+   ],
+   "offset": 268611072,
+   "byteLength": 7077888
+  },
+  {
+   "name": "block6.o",
+   "shape": [
+    768,
+    768
+   ],
+   "offset": 275688960,
+   "byteLength": 2359296
+  },
+  {
+   "name": "block6.ln_ff",
+   "shape": [
+    768
+   ],
+   "offset": 278048256,
+   "byteLength": 3072
+  },
+  {
+   "name": "block6.wi",
+   "shape": [
+    4096,
+    768
+   ],
+   "offset": 278051328,
+   "byteLength": 12582912
+  },
+  {
+   "name": "block6.wo",
+   "shape": [
+    768,
+    2048
+   ],
+   "offset": 290634240,
+   "byteLength": 6291456
+  },
+  {
+   "name": "block7.ln_attn",
+   "shape": [
+    768
+   ],
+   "offset": 296925696,
+   "byteLength": 3072
+  },
+  {
+   "name": "block7.qkv",
+   "shape": [
+    2304,
+    768
+   ],
+   "offset": 296928768,
+   "byteLength": 7077888
+  },
+  {
+   "name": "block7.o",
+   "shape": [
+    768,
+    768
+   ],
+   "offset": 304006656,
+   "byteLength": 2359296
+  },
+  {
+   "name": "block7.ln_ff",
+   "shape": [
+    768
+   ],
+   "offset": 306365952,
+   "byteLength": 3072
+  },
+  {
+   "name": "block7.wi",
+   "shape": [
+    4096,
+    768
+   ],
+   "offset": 306369024,
+   "byteLength": 12582912
+  },
+  {
+   "name": "block7.wo",
+   "shape": [
+    768,
+    2048
+   ],
+   "offset": 318951936,
+   "byteLength": 6291456
+  },
+  {
+   "name": "block8.ln_attn",
+   "shape": [
+    768
+   ],
+   "offset": 325243392,
+   "byteLength": 3072
+  },
+  {
+   "name": "block8.qkv",
+   "shape": [
+    2304,
+    768
+   ],
+   "offset": 325246464,
+   "byteLength": 7077888
+  },
+  {
+   "name": "block8.o",
+   "shape": [
+    768,
+    768
+   ],
+   "offset": 332324352,
+   "byteLength": 2359296
+  },
+  {
+   "name": "block8.ln_ff",
+   "shape": [
+    768
+   ],
+   "offset": 334683648,
+   "byteLength": 3072
+  },
+  {
+   "name": "block8.wi",
+   "shape": [
+    4096,
+    768
+   ],
+   "offset": 334686720,
+   "byteLength": 12582912
+  },
+  {
+   "name": "block8.wo",
+   "shape": [
+    768,
+    2048
+   ],
+   "offset": 347269632,
+   "byteLength": 6291456
+  },
+  {
+   "name": "block9.ln_attn",
+   "shape": [
+    768
+   ],
+   "offset": 353561088,
+   "byteLength": 3072
+  },
+  {
+   "name": "block9.qkv",
+   "shape": [
+    2304,
+    768
+   ],
+   "offset": 353564160,
+   "byteLength": 7077888
+  },
+  {
+   "name": "block9.o",
+   "shape": [
+    768,
+    768
+   ],
+   "offset": 360642048,
+   "byteLength": 2359296
+  },
+  {
+   "name": "block9.ln_ff",
+   "shape": [
+    768
+   ],
+   "offset": 363001344,
+   "byteLength": 3072
+  },
+  {
+   "name": "block9.wi",
+   "shape": [
+    4096,
+    768
+   ],
+   "offset": 363004416,
+   "byteLength": 12582912
+  },
+  {
+   "name": "block9.wo",
+   "shape": [
+    768,
+    2048
+   ],
+   "offset": 375587328,
+   "byteLength": 6291456
+  },
+  {
+   "name": "block10.ln_attn",
+   "shape": [
+    768
+   ],
+   "offset": 381878784,
+   "byteLength": 3072
+  },
+  {
+   "name": "block10.qkv",
+   "shape": [
+    2304,
+    768
+   ],
+   "offset": 381881856,
+   "byteLength": 7077888
+  },
+  {
+   "name": "block10.o",
+   "shape": [
+    768,
+    768
+   ],
+   "offset": 388959744,
+   "byteLength": 2359296
+  },
+  {
+   "name": "block10.ln_ff",
+   "shape": [
+    768
+   ],
+   "offset": 391319040,
+   "byteLength": 3072
+  },
+  {
+   "name": "block10.wi",
+   "shape": [
+    4096,
+    768
+   ],
+   "offset": 391322112,
+   "byteLength": 12582912
+  },
+  {
+   "name": "block10.wo",
+   "shape": [
+    768,
+    2048
+   ],
+   "offset": 403905024,
+   "byteLength": 6291456
+  },
+  {
+   "name": "block11.ln_attn",
+   "shape": [
+    768
+   ],
+   "offset": 410196480,
+   "byteLength": 3072
+  },
+  {
+   "name": "block11.qkv",
+   "shape": [
+    2304,
+    768
+   ],
+   "offset": 410199552,
+   "byteLength": 7077888
+  },
+  {
+   "name": "block11.o",
+   "shape": [
+    768,
+    768
+   ],
+   "offset": 417277440,
+   "byteLength": 2359296
+  },
+  {
+   "name": "block11.ln_ff",
+   "shape": [
+    768
+   ],
+   "offset": 419636736,
+   "byteLength": 3072
+  },
+  {
+   "name": "block11.wi",
+   "shape": [
+    4096,
+    768
+   ],
+   "offset": 419639808,
+   "byteLength": 12582912
+  },
+  {
+   "name": "block11.wo",
+   "shape": [
+    768,
+    2048
+   ],
+   "offset": 432222720,
+   "byteLength": 6291456
+  }
+ ],
+ "relative_buckets": {
+  "min_distance": -511,
+  "buckets": [
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   15,
+   14,
+   14,
+   14,
+   14,
+   14,
+   14,
+   14,
+   14,
+   14,
+   14,
+   14,
+   14,
+   14,
+   14,
+   14,
+   14,
+   14,
+   14,
+   14,
+   14,
+   14,
+   14,
+   14,
+   14,
+   14,
+   14,
+   14,
+   13,
+   13,
+   13,
+   13,
+   13,
+   13,
+   13,
+   13,
+   13,
+   13,
+   13,
+   13,
+   13,
+   13,
+   13,
+   13,
+   13,
+   13,
+   12,
+   12,
+   12,
+   12,
+   12,
+   12,
+   12,
+   12,
+   12,
+   12,
+   12,
+   12,
+   12,
+   12,
+   11,
+   11,
+   11,
+   11,
+   11,
+   11,
+   11,
+   11,
+   11,
+   10,
+   10,
+   10,
+   10,
+   10,
+   10,
+   10,
+   9,
+   9,
+   9,
+   9,
+   8,
+   8,
+   8,
+   8,
+   7,
+   6,
+   5,
+   4,
+   3,
+   2,
+   1,
+   0,
+   17,
+   18,
+   19,
+   20,
+   21,
+   22,
+   23,
+   24,
+   24,
+   24,
+   24,
+   25,
+   25,
+   25,
+   25,
+   26,
+   26,
+   26,
+   26,
+   26,
+   26,
+   26,
+   27,
+   27,
+   27,
+   27,
+   27,
+   27,
+   27,
+   27,
+   27,
+   28,
+   28,
+   28,
+   28,
+   28,
+   28,
+   28,
+   28,
+   28,
+   28,
+   28,
+   28,
+   28,
+   28,
+   29,
+   29,
+   29,
+   29,
+   29,
+   29,
+   29,
+   29,
+   29,
+   29,
+   29,
+   29,
+   29,
+   29,
+   29,
+   29,
+   29,
+   29,
+   30,
+   30,
+   30,
+   30,
+   30,
+   30,
+   30,
+   30,
+   30,
+   30,
+   30,
+   30,
+   30,
+   30,
+   30,
+   30,
+   30,
+   30,
+   30,
+   30,
+   30,
+   30,
+   30,
+   30,
+   30,
+   30,
+   30,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31,
+   31
+  ]
+ }
+}

unimate/manifest.json

@@ -0,0 +1,4227 @@
+{
+ "schema": "kaminos.unimate.denoiser-weights.v0",
+ "file": "unimate-denoiser-fp32.bin",
+ "byteLength": 297891584,
+ "sha256": "4dc5d6953fd120842e39c808931c13794fe0abedc5597f1031f9393bd8b5285b",
+ "alignment": 256,
+ "parameterCount": 74469840,
+ "source": {
+  "unimate_rev": "b78c7807df7cf4231d4e67f8d0ece29c5e11025b",
+  "checkpoint": "Linzhan/UniMate unimate_uniml3d_f60_v3/checkpoints/checkpoint_step_150000.pt (EMA)",
+  "checkpoint_sha256": "f991ddd87262c251f4681d18297a6163ab4f139daa28c86d5ef0a66771df7bc6",
+  "config_sha256": "c8d3b44ee5ea6c5bd71e7b5ad1f4c94019314a71ac2578d59352618b6342374c",
+  "ema_applied_by": "unimate.inference.sample._load_checkpoint",
+  "weights_license": "CC-BY-NC-4.0 (released UniMate weights)"
+ },
+ "model": {
+  "attention": "graph",
+  "text_cond": "adaln",
+  "latent_dim": 512,
+  "ff_size": 2048,
+  "num_layers": 10,
+  "num_heads": 8,
+  "dropout": 0,
+  "use_spectral_rope": true,
+  "max_freqs": 8,
+  "use_signnet": true,
+  "cond_mode": "text",
+  "cond_mask_prob": 0.1,
+  "use_joint_name_emb": true,
+  "use_graph_emb": false,
+  "use_depth_emb": true,
+  "concat_parent_features": true,
+  "num_tpos_queries": 4,
+  "inject_tpos_to_adaln": true,
+  "use_graph_attn_bias": true,
+  "share_graph_attn_bias": false,
+  "gradient_checkpointing": false,
+  "text_encoder_type": "t5",
+  "text_encoder_version": "google/flan-t5-base"
+ },
+ "dataset": {
+  "max_joints": 71,
+  "max_motion_length": 60,
+  "feature_len": 12,
+  "max_depth": 20
+ },
+ "tensors": [
+  {
+   "name": "cond_embedder.weight",
+   "shape": [
+    512,
+    768
+   ],
+   "dtype": "float32",
+   "offset": 0,
+   "byteLength": 1572864
+  },
+  {
+   "name": "cond_embedder.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 1572864,
+   "byteLength": 2048
+  },
+  {
+   "name": "time_embedder.mlp.0.weight",
+   "shape": [
+    512,
+    256
+   ],
+   "dtype": "float32",
+   "offset": 1574912,
+   "byteLength": 524288
+  },
+  {
+   "name": "time_embedder.mlp.0.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 2099200,
+   "byteLength": 2048
+  },
+  {
+   "name": "time_embedder.mlp.2.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 2101248,
+   "byteLength": 1048576
+  },
+  {
+   "name": "time_embedder.mlp.2.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 3149824,
+   "byteLength": 2048
+  },
+  {
+   "name": "input_layer.root_tpos_embedder.0.weight",
+   "shape": [
+    512,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 3151872,
+   "byteLength": 24576
+  },
+  {
+   "name": "input_layer.root_tpos_embedder.0.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 3176448,
+   "byteLength": 2048
+  },
+  {
+   "name": "input_layer.root_tpos_embedder.2.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 3178496,
+   "byteLength": 1048576
+  },
+  {
+   "name": "input_layer.root_tpos_embedder.2.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 4227072,
+   "byteLength": 2048
+  },
+  {
+   "name": "input_layer.root_x_embedder.0.weight",
+   "shape": [
+    512,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 4229120,
+   "byteLength": 24576
+  },
+  {
+   "name": "input_layer.root_x_embedder.0.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 4253696,
+   "byteLength": 2048
+  },
+  {
+   "name": "input_layer.root_x_embedder.2.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 4255744,
+   "byteLength": 1048576
+  },
+  {
+   "name": "input_layer.root_x_embedder.2.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 5304320,
+   "byteLength": 2048
+  },
+  {
+   "name": "input_layer.joint_tpos_embedder.0.weight",
+   "shape": [
+    512,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 5306368,
+   "byteLength": 24576
+  },
+  {
+   "name": "input_layer.joint_tpos_embedder.0.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 5330944,
+   "byteLength": 2048
+  },
+  {
+   "name": "input_layer.joint_tpos_embedder.2.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 5332992,
+   "byteLength": 1048576
+  },
+  {
+   "name": "input_layer.joint_tpos_embedder.2.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 6381568,
+   "byteLength": 2048
+  },
+  {
+   "name": "input_layer.parent_tpos_embedder.0.weight",
+   "shape": [
+    512,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 6383616,
+   "byteLength": 24576
+  },
+  {
+   "name": "input_layer.parent_tpos_embedder.0.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 6408192,
+   "byteLength": 2048
+  },
+  {
+   "name": "input_layer.parent_tpos_embedder.2.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 6410240,
+   "byteLength": 1048576
+  },
+  {
+   "name": "input_layer.parent_tpos_embedder.2.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 7458816,
+   "byteLength": 2048
+  },
+  {
+   "name": "input_layer.tpos_fuse.0.weight",
+   "shape": [
+    512,
+    1024
+   ],
+   "dtype": "float32",
+   "offset": 7460864,
+   "byteLength": 2097152
+  },
+  {
+   "name": "input_layer.tpos_fuse.0.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 9558016,
+   "byteLength": 2048
+  },
+  {
+   "name": "input_layer.tpos_fuse.2.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 9560064,
+   "byteLength": 1048576
+  },
+  {
+   "name": "input_layer.tpos_fuse.2.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 10608640,
+   "byteLength": 2048
+  },
+  {
+   "name": "input_layer.joint_x_embedder.0.weight",
+   "shape": [
+    512,
+    12
+   ],
+   "dtype": "float32",
+   "offset": 10610688,
+   "byteLength": 24576
+  },
+  {
+   "name": "input_layer.joint_x_embedder.0.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 10635264,
+   "byteLength": 2048
+  },
+  {
+   "name": "input_layer.joint_x_embedder.2.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 10637312,
+   "byteLength": 1048576
+  },
+  {
+   "name": "input_layer.joint_x_embedder.2.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 11685888,
+   "byteLength": 2048
+  },
+  {
+   "name": "tpos_pool.pool.queries",
+   "shape": [
+    1,
+    4,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 11687936,
+   "byteLength": 8192
+  },
+  {
+   "name": "tpos_pool.pool.norm_q.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 11696128,
+   "byteLength": 2048
+  },
+  {
+   "name": "tpos_pool.pool.norm_kv.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 11698176,
+   "byteLength": 2048
+  },
+  {
+   "name": "tpos_pool.pool.cross_attn.in_proj_weight",
+   "shape": [
+    1536,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 11700224,
+   "byteLength": 3145728
+  },
+  {
+   "name": "tpos_pool.pool.cross_attn.in_proj_bias",
+   "shape": [
+    1536
+   ],
+   "dtype": "float32",
+   "offset": 14845952,
+   "byteLength": 6144
+  },
+  {
+   "name": "tpos_pool.pool.cross_attn.out_proj.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 14852096,
+   "byteLength": 1048576
+  },
+  {
+   "name": "tpos_pool.pool.cross_attn.out_proj.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 15900672,
+   "byteLength": 2048
+  },
+  {
+   "name": "tpos_pool.pool.norm_ff.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 15902720,
+   "byteLength": 2048
+  },
+  {
+   "name": "tpos_pool.pool.ffn.w12.weight",
+   "shape": [
+    2048,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 15904768,
+   "byteLength": 4194304
+  },
+  {
+   "name": "tpos_pool.pool.ffn.w12.bias",
+   "shape": [
+    2048
+   ],
+   "dtype": "float32",
+   "offset": 20099072,
+   "byteLength": 8192
+  },
+  {
+   "name": "tpos_pool.pool.ffn.w3.weight",
+   "shape": [
+    512,
+    1024
+   ],
+   "dtype": "float32",
+   "offset": 20107264,
+   "byteLength": 2097152
+  },
+  {
+   "name": "tpos_pool.pool.ffn.w3.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 22204416,
+   "byteLength": 2048
+  },
+  {
+   "name": "tpos_pool.pool.norm_out.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 22206464,
+   "byteLength": 2048
+  },
+  {
+   "name": "tpos_pool.pool.out_proj.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 22208512,
+   "byteLength": 1048576
+  },
+  {
+   "name": "tpos_pool.pool.out_proj.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 23257088,
+   "byteLength": 2048
+  },
+  {
+   "name": "joint_name_embedder.weight",
+   "shape": [
+    512,
+    768
+   ],
+   "dtype": "float32",
+   "offset": 23259136,
+   "byteLength": 1572864
+  },
+  {
+   "name": "joint_name_embedder.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 24832000,
+   "byteLength": 2048
+  },
+  {
+   "name": "depth_embedding.weight",
+   "shape": [
+    21,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 24834048,
+   "byteLength": 43008
+  },
+  {
+   "name": "rope_j.spectral_encoder.phi.0.weight",
+   "shape": [
+    64,
+    1
+   ],
+   "dtype": "float32",
+   "offset": 24877056,
+   "byteLength": 256
+  },
+  {
+   "name": "rope_j.spectral_encoder.phi.0.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 24877312,
+   "byteLength": 256
+  },
+  {
+   "name": "rope_j.spectral_encoder.phi.2.weight",
+   "shape": [
+    64,
+    64
+   ],
+   "dtype": "float32",
+   "offset": 24877568,
+   "byteLength": 16384
+  },
+  {
+   "name": "rope_j.spectral_encoder.phi.2.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 24893952,
+   "byteLength": 256
+  },
+  {
+   "name": "rope_j.spectral_encoder.rho.0.weight",
+   "shape": [
+    64,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 24894208,
+   "byteLength": 131072
+  },
+  {
+   "name": "rope_j.spectral_encoder.rho.0.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 25025280,
+   "byteLength": 256
+  },
+  {
+   "name": "rope_j.spectral_encoder.rho.2.weight",
+   "shape": [
+    32,
+    64
+   ],
+   "dtype": "float32",
+   "offset": 25025536,
+   "byteLength": 8192
+  },
+  {
+   "name": "rope_j.spectral_encoder.rho.2.bias",
+   "shape": [
+    32
+   ],
+   "dtype": "float32",
+   "offset": 25033728,
+   "byteLength": 128
+  },
+  {
+   "name": "transformer_blocks.0.norm_s.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 25033984,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.0.s_attn.graph_dist_scale",
+   "shape": [
+    1
+   ],
+   "dtype": "float32",
+   "offset": 25036032,
+   "byteLength": 4
+  },
+  {
+   "name": "transformer_blocks.0.s_attn.graph_rel_scale",
+   "shape": [
+    1
+   ],
+   "dtype": "float32",
+   "offset": 25036288,
+   "byteLength": 4
+  },
+  {
+   "name": "transformer_blocks.0.s_attn.rope.spectral_encoder.phi.0.weight",
+   "shape": [
+    64,
+    1
+   ],
+   "dtype": "float32",
+   "offset": 25036544,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.0.s_attn.rope.spectral_encoder.phi.0.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 25036800,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.0.s_attn.rope.spectral_encoder.phi.2.weight",
+   "shape": [
+    64,
+    64
+   ],
+   "dtype": "float32",
+   "offset": 25037056,
+   "byteLength": 16384
+  },
+  {
+   "name": "transformer_blocks.0.s_attn.rope.spectral_encoder.phi.2.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 25053440,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.0.s_attn.rope.spectral_encoder.rho.0.weight",
+   "shape": [
+    64,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 25053696,
+   "byteLength": 131072
+  },
+  {
+   "name": "transformer_blocks.0.s_attn.rope.spectral_encoder.rho.0.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 25184768,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.0.s_attn.rope.spectral_encoder.rho.2.weight",
+   "shape": [
+    32,
+    64
+   ],
+   "dtype": "float32",
+   "offset": 25185024,
+   "byteLength": 8192
+  },
+  {
+   "name": "transformer_blocks.0.s_attn.rope.spectral_encoder.rho.2.bias",
+   "shape": [
+    32
+   ],
+   "dtype": "float32",
+   "offset": 25193216,
+   "byteLength": 128
+  },
+  {
+   "name": "transformer_blocks.0.s_attn.qkv.weight",
+   "shape": [
+    1536,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 25193472,
+   "byteLength": 3145728
+  },
+  {
+   "name": "transformer_blocks.0.s_attn.qkv.bias",
+   "shape": [
+    1536
+   ],
+   "dtype": "float32",
+   "offset": 28339200,
+   "byteLength": 6144
+  },
+  {
+   "name": "transformer_blocks.0.s_attn.proj.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 28345344,
+   "byteLength": 1048576
+  },
+  {
+   "name": "transformer_blocks.0.s_attn.proj.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 29393920,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.0.s_attn.q_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 29395968,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.0.s_attn.k_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 29396224,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.0.s_attn.graph_dist_embedding.weight",
+   "shape": [
+    6,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 29396480,
+   "byteLength": 3072
+  },
+  {
+   "name": "transformer_blocks.0.s_attn.graph_dist_proj.weight",
+   "shape": [
+    8,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 29399552,
+   "byteLength": 4096
+  },
+  {
+   "name": "transformer_blocks.0.s_attn.graph_dist_proj.bias",
+   "shape": [
+    8
+   ],
+   "dtype": "float32",
+   "offset": 29403648,
+   "byteLength": 32
+  },
+  {
+   "name": "transformer_blocks.0.s_attn.graph_rel_embedding.weight",
+   "shape": [
+    6,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 29403904,
+   "byteLength": 3072
+  },
+  {
+   "name": "transformer_blocks.0.s_attn.graph_rel_proj.weight",
+   "shape": [
+    8,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 29406976,
+   "byteLength": 4096
+  },
+  {
+   "name": "transformer_blocks.0.s_attn.graph_rel_proj.bias",
+   "shape": [
+    8
+   ],
+   "dtype": "float32",
+   "offset": 29411072,
+   "byteLength": 32
+  },
+  {
+   "name": "transformer_blocks.0.norm_t.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 29411328,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.0.t_attn.qkv.weight",
+   "shape": [
+    1536,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 29413376,
+   "byteLength": 3145728
+  },
+  {
+   "name": "transformer_blocks.0.t_attn.qkv.bias",
+   "shape": [
+    1536
+   ],
+   "dtype": "float32",
+   "offset": 32559104,
+   "byteLength": 6144
+  },
+  {
+   "name": "transformer_blocks.0.t_attn.proj.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 32565248,
+   "byteLength": 1048576
+  },
+  {
+   "name": "transformer_blocks.0.t_attn.proj.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 33613824,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.0.t_attn.q_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 33615872,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.0.t_attn.k_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 33616128,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.0.norm_mlp.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 33616384,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.0.mlp.w12.weight",
+   "shape": [
+    2730,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 33618432,
+   "byteLength": 5591040
+  },
+  {
+   "name": "transformer_blocks.0.mlp.w12.bias",
+   "shape": [
+    2730
+   ],
+   "dtype": "float32",
+   "offset": 39209472,
+   "byteLength": 10920
+  },
+  {
+   "name": "transformer_blocks.0.mlp.w3.weight",
+   "shape": [
+    512,
+    1365
+   ],
+   "dtype": "float32",
+   "offset": 39220480,
+   "byteLength": 2795520
+  },
+  {
+   "name": "transformer_blocks.0.mlp.w3.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 42016000,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.0.adaLN_modulation.1.weight",
+   "shape": [
+    4608,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 42018048,
+   "byteLength": 9437184
+  },
+  {
+   "name": "transformer_blocks.0.adaLN_modulation.1.bias",
+   "shape": [
+    4608
+   ],
+   "dtype": "float32",
+   "offset": 51455232,
+   "byteLength": 18432
+  },
+  {
+   "name": "transformer_blocks.1.norm_s.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 51473664,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.1.s_attn.graph_dist_scale",
+   "shape": [
+    1
+   ],
+   "dtype": "float32",
+   "offset": 51475712,
+   "byteLength": 4
+  },
+  {
+   "name": "transformer_blocks.1.s_attn.graph_rel_scale",
+   "shape": [
+    1
+   ],
+   "dtype": "float32",
+   "offset": 51475968,
+   "byteLength": 4
+  },
+  {
+   "name": "transformer_blocks.1.s_attn.rope.spectral_encoder.phi.0.weight",
+   "shape": [
+    64,
+    1
+   ],
+   "dtype": "float32",
+   "offset": 51476224,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.1.s_attn.rope.spectral_encoder.phi.0.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 51476480,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.1.s_attn.rope.spectral_encoder.phi.2.weight",
+   "shape": [
+    64,
+    64
+   ],
+   "dtype": "float32",
+   "offset": 51476736,
+   "byteLength": 16384
+  },
+  {
+   "name": "transformer_blocks.1.s_attn.rope.spectral_encoder.phi.2.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 51493120,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.1.s_attn.rope.spectral_encoder.rho.0.weight",
+   "shape": [
+    64,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 51493376,
+   "byteLength": 131072
+  },
+  {
+   "name": "transformer_blocks.1.s_attn.rope.spectral_encoder.rho.0.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 51624448,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.1.s_attn.rope.spectral_encoder.rho.2.weight",
+   "shape": [
+    32,
+    64
+   ],
+   "dtype": "float32",
+   "offset": 51624704,
+   "byteLength": 8192
+  },
+  {
+   "name": "transformer_blocks.1.s_attn.rope.spectral_encoder.rho.2.bias",
+   "shape": [
+    32
+   ],
+   "dtype": "float32",
+   "offset": 51632896,
+   "byteLength": 128
+  },
+  {
+   "name": "transformer_blocks.1.s_attn.qkv.weight",
+   "shape": [
+    1536,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 51633152,
+   "byteLength": 3145728
+  },
+  {
+   "name": "transformer_blocks.1.s_attn.qkv.bias",
+   "shape": [
+    1536
+   ],
+   "dtype": "float32",
+   "offset": 54778880,
+   "byteLength": 6144
+  },
+  {
+   "name": "transformer_blocks.1.s_attn.proj.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 54785024,
+   "byteLength": 1048576
+  },
+  {
+   "name": "transformer_blocks.1.s_attn.proj.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 55833600,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.1.s_attn.q_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 55835648,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.1.s_attn.k_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 55835904,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.1.s_attn.graph_dist_embedding.weight",
+   "shape": [
+    6,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 55836160,
+   "byteLength": 3072
+  },
+  {
+   "name": "transformer_blocks.1.s_attn.graph_dist_proj.weight",
+   "shape": [
+    8,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 55839232,
+   "byteLength": 4096
+  },
+  {
+   "name": "transformer_blocks.1.s_attn.graph_dist_proj.bias",
+   "shape": [
+    8
+   ],
+   "dtype": "float32",
+   "offset": 55843328,
+   "byteLength": 32
+  },
+  {
+   "name": "transformer_blocks.1.s_attn.graph_rel_embedding.weight",
+   "shape": [
+    6,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 55843584,
+   "byteLength": 3072
+  },
+  {
+   "name": "transformer_blocks.1.s_attn.graph_rel_proj.weight",
+   "shape": [
+    8,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 55846656,
+   "byteLength": 4096
+  },
+  {
+   "name": "transformer_blocks.1.s_attn.graph_rel_proj.bias",
+   "shape": [
+    8
+   ],
+   "dtype": "float32",
+   "offset": 55850752,
+   "byteLength": 32
+  },
+  {
+   "name": "transformer_blocks.1.norm_t.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 55851008,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.1.t_attn.qkv.weight",
+   "shape": [
+    1536,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 55853056,
+   "byteLength": 3145728
+  },
+  {
+   "name": "transformer_blocks.1.t_attn.qkv.bias",
+   "shape": [
+    1536
+   ],
+   "dtype": "float32",
+   "offset": 58998784,
+   "byteLength": 6144
+  },
+  {
+   "name": "transformer_blocks.1.t_attn.proj.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 59004928,
+   "byteLength": 1048576
+  },
+  {
+   "name": "transformer_blocks.1.t_attn.proj.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 60053504,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.1.t_attn.q_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 60055552,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.1.t_attn.k_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 60055808,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.1.norm_mlp.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 60056064,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.1.mlp.w12.weight",
+   "shape": [
+    2730,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 60058112,
+   "byteLength": 5591040
+  },
+  {
+   "name": "transformer_blocks.1.mlp.w12.bias",
+   "shape": [
+    2730
+   ],
+   "dtype": "float32",
+   "offset": 65649152,
+   "byteLength": 10920
+  },
+  {
+   "name": "transformer_blocks.1.mlp.w3.weight",
+   "shape": [
+    512,
+    1365
+   ],
+   "dtype": "float32",
+   "offset": 65660160,
+   "byteLength": 2795520
+  },
+  {
+   "name": "transformer_blocks.1.mlp.w3.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 68455680,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.1.adaLN_modulation.1.weight",
+   "shape": [
+    4608,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 68457728,
+   "byteLength": 9437184
+  },
+  {
+   "name": "transformer_blocks.1.adaLN_modulation.1.bias",
+   "shape": [
+    4608
+   ],
+   "dtype": "float32",
+   "offset": 77894912,
+   "byteLength": 18432
+  },
+  {
+   "name": "transformer_blocks.2.norm_s.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 77913344,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.2.s_attn.graph_dist_scale",
+   "shape": [
+    1
+   ],
+   "dtype": "float32",
+   "offset": 77915392,
+   "byteLength": 4
+  },
+  {
+   "name": "transformer_blocks.2.s_attn.graph_rel_scale",
+   "shape": [
+    1
+   ],
+   "dtype": "float32",
+   "offset": 77915648,
+   "byteLength": 4
+  },
+  {
+   "name": "transformer_blocks.2.s_attn.rope.spectral_encoder.phi.0.weight",
+   "shape": [
+    64,
+    1
+   ],
+   "dtype": "float32",
+   "offset": 77915904,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.2.s_attn.rope.spectral_encoder.phi.0.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 77916160,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.2.s_attn.rope.spectral_encoder.phi.2.weight",
+   "shape": [
+    64,
+    64
+   ],
+   "dtype": "float32",
+   "offset": 77916416,
+   "byteLength": 16384
+  },
+  {
+   "name": "transformer_blocks.2.s_attn.rope.spectral_encoder.phi.2.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 77932800,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.2.s_attn.rope.spectral_encoder.rho.0.weight",
+   "shape": [
+    64,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 77933056,
+   "byteLength": 131072
+  },
+  {
+   "name": "transformer_blocks.2.s_attn.rope.spectral_encoder.rho.0.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 78064128,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.2.s_attn.rope.spectral_encoder.rho.2.weight",
+   "shape": [
+    32,
+    64
+   ],
+   "dtype": "float32",
+   "offset": 78064384,
+   "byteLength": 8192
+  },
+  {
+   "name": "transformer_blocks.2.s_attn.rope.spectral_encoder.rho.2.bias",
+   "shape": [
+    32
+   ],
+   "dtype": "float32",
+   "offset": 78072576,
+   "byteLength": 128
+  },
+  {
+   "name": "transformer_blocks.2.s_attn.qkv.weight",
+   "shape": [
+    1536,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 78072832,
+   "byteLength": 3145728
+  },
+  {
+   "name": "transformer_blocks.2.s_attn.qkv.bias",
+   "shape": [
+    1536
+   ],
+   "dtype": "float32",
+   "offset": 81218560,
+   "byteLength": 6144
+  },
+  {
+   "name": "transformer_blocks.2.s_attn.proj.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 81224704,
+   "byteLength": 1048576
+  },
+  {
+   "name": "transformer_blocks.2.s_attn.proj.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 82273280,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.2.s_attn.q_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 82275328,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.2.s_attn.k_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 82275584,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.2.s_attn.graph_dist_embedding.weight",
+   "shape": [
+    6,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 82275840,
+   "byteLength": 3072
+  },
+  {
+   "name": "transformer_blocks.2.s_attn.graph_dist_proj.weight",
+   "shape": [
+    8,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 82278912,
+   "byteLength": 4096
+  },
+  {
+   "name": "transformer_blocks.2.s_attn.graph_dist_proj.bias",
+   "shape": [
+    8
+   ],
+   "dtype": "float32",
+   "offset": 82283008,
+   "byteLength": 32
+  },
+  {
+   "name": "transformer_blocks.2.s_attn.graph_rel_embedding.weight",
+   "shape": [
+    6,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 82283264,
+   "byteLength": 3072
+  },
+  {
+   "name": "transformer_blocks.2.s_attn.graph_rel_proj.weight",
+   "shape": [
+    8,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 82286336,
+   "byteLength": 4096
+  },
+  {
+   "name": "transformer_blocks.2.s_attn.graph_rel_proj.bias",
+   "shape": [
+    8
+   ],
+   "dtype": "float32",
+   "offset": 82290432,
+   "byteLength": 32
+  },
+  {
+   "name": "transformer_blocks.2.norm_t.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 82290688,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.2.t_attn.qkv.weight",
+   "shape": [
+    1536,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 82292736,
+   "byteLength": 3145728
+  },
+  {
+   "name": "transformer_blocks.2.t_attn.qkv.bias",
+   "shape": [
+    1536
+   ],
+   "dtype": "float32",
+   "offset": 85438464,
+   "byteLength": 6144
+  },
+  {
+   "name": "transformer_blocks.2.t_attn.proj.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 85444608,
+   "byteLength": 1048576
+  },
+  {
+   "name": "transformer_blocks.2.t_attn.proj.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 86493184,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.2.t_attn.q_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 86495232,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.2.t_attn.k_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 86495488,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.2.norm_mlp.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 86495744,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.2.mlp.w12.weight",
+   "shape": [
+    2730,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 86497792,
+   "byteLength": 5591040
+  },
+  {
+   "name": "transformer_blocks.2.mlp.w12.bias",
+   "shape": [
+    2730
+   ],
+   "dtype": "float32",
+   "offset": 92088832,
+   "byteLength": 10920
+  },
+  {
+   "name": "transformer_blocks.2.mlp.w3.weight",
+   "shape": [
+    512,
+    1365
+   ],
+   "dtype": "float32",
+   "offset": 92099840,
+   "byteLength": 2795520
+  },
+  {
+   "name": "transformer_blocks.2.mlp.w3.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 94895360,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.2.adaLN_modulation.1.weight",
+   "shape": [
+    4608,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 94897408,
+   "byteLength": 9437184
+  },
+  {
+   "name": "transformer_blocks.2.adaLN_modulation.1.bias",
+   "shape": [
+    4608
+   ],
+   "dtype": "float32",
+   "offset": 104334592,
+   "byteLength": 18432
+  },
+  {
+   "name": "transformer_blocks.3.norm_s.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 104353024,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.3.s_attn.graph_dist_scale",
+   "shape": [
+    1
+   ],
+   "dtype": "float32",
+   "offset": 104355072,
+   "byteLength": 4
+  },
+  {
+   "name": "transformer_blocks.3.s_attn.graph_rel_scale",
+   "shape": [
+    1
+   ],
+   "dtype": "float32",
+   "offset": 104355328,
+   "byteLength": 4
+  },
+  {
+   "name": "transformer_blocks.3.s_attn.rope.spectral_encoder.phi.0.weight",
+   "shape": [
+    64,
+    1
+   ],
+   "dtype": "float32",
+   "offset": 104355584,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.3.s_attn.rope.spectral_encoder.phi.0.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 104355840,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.3.s_attn.rope.spectral_encoder.phi.2.weight",
+   "shape": [
+    64,
+    64
+   ],
+   "dtype": "float32",
+   "offset": 104356096,
+   "byteLength": 16384
+  },
+  {
+   "name": "transformer_blocks.3.s_attn.rope.spectral_encoder.phi.2.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 104372480,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.3.s_attn.rope.spectral_encoder.rho.0.weight",
+   "shape": [
+    64,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 104372736,
+   "byteLength": 131072
+  },
+  {
+   "name": "transformer_blocks.3.s_attn.rope.spectral_encoder.rho.0.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 104503808,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.3.s_attn.rope.spectral_encoder.rho.2.weight",
+   "shape": [
+    32,
+    64
+   ],
+   "dtype": "float32",
+   "offset": 104504064,
+   "byteLength": 8192
+  },
+  {
+   "name": "transformer_blocks.3.s_attn.rope.spectral_encoder.rho.2.bias",
+   "shape": [
+    32
+   ],
+   "dtype": "float32",
+   "offset": 104512256,
+   "byteLength": 128
+  },
+  {
+   "name": "transformer_blocks.3.s_attn.qkv.weight",
+   "shape": [
+    1536,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 104512512,
+   "byteLength": 3145728
+  },
+  {
+   "name": "transformer_blocks.3.s_attn.qkv.bias",
+   "shape": [
+    1536
+   ],
+   "dtype": "float32",
+   "offset": 107658240,
+   "byteLength": 6144
+  },
+  {
+   "name": "transformer_blocks.3.s_attn.proj.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 107664384,
+   "byteLength": 1048576
+  },
+  {
+   "name": "transformer_blocks.3.s_attn.proj.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 108712960,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.3.s_attn.q_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 108715008,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.3.s_attn.k_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 108715264,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.3.s_attn.graph_dist_embedding.weight",
+   "shape": [
+    6,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 108715520,
+   "byteLength": 3072
+  },
+  {
+   "name": "transformer_blocks.3.s_attn.graph_dist_proj.weight",
+   "shape": [
+    8,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 108718592,
+   "byteLength": 4096
+  },
+  {
+   "name": "transformer_blocks.3.s_attn.graph_dist_proj.bias",
+   "shape": [
+    8
+   ],
+   "dtype": "float32",
+   "offset": 108722688,
+   "byteLength": 32
+  },
+  {
+   "name": "transformer_blocks.3.s_attn.graph_rel_embedding.weight",
+   "shape": [
+    6,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 108722944,
+   "byteLength": 3072
+  },
+  {
+   "name": "transformer_blocks.3.s_attn.graph_rel_proj.weight",
+   "shape": [
+    8,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 108726016,
+   "byteLength": 4096
+  },
+  {
+   "name": "transformer_blocks.3.s_attn.graph_rel_proj.bias",
+   "shape": [
+    8
+   ],
+   "dtype": "float32",
+   "offset": 108730112,
+   "byteLength": 32
+  },
+  {
+   "name": "transformer_blocks.3.norm_t.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 108730368,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.3.t_attn.qkv.weight",
+   "shape": [
+    1536,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 108732416,
+   "byteLength": 3145728
+  },
+  {
+   "name": "transformer_blocks.3.t_attn.qkv.bias",
+   "shape": [
+    1536
+   ],
+   "dtype": "float32",
+   "offset": 111878144,
+   "byteLength": 6144
+  },
+  {
+   "name": "transformer_blocks.3.t_attn.proj.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 111884288,
+   "byteLength": 1048576
+  },
+  {
+   "name": "transformer_blocks.3.t_attn.proj.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 112932864,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.3.t_attn.q_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 112934912,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.3.t_attn.k_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 112935168,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.3.norm_mlp.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 112935424,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.3.mlp.w12.weight",
+   "shape": [
+    2730,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 112937472,
+   "byteLength": 5591040
+  },
+  {
+   "name": "transformer_blocks.3.mlp.w12.bias",
+   "shape": [
+    2730
+   ],
+   "dtype": "float32",
+   "offset": 118528512,
+   "byteLength": 10920
+  },
+  {
+   "name": "transformer_blocks.3.mlp.w3.weight",
+   "shape": [
+    512,
+    1365
+   ],
+   "dtype": "float32",
+   "offset": 118539520,
+   "byteLength": 2795520
+  },
+  {
+   "name": "transformer_blocks.3.mlp.w3.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 121335040,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.3.adaLN_modulation.1.weight",
+   "shape": [
+    4608,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 121337088,
+   "byteLength": 9437184
+  },
+  {
+   "name": "transformer_blocks.3.adaLN_modulation.1.bias",
+   "shape": [
+    4608
+   ],
+   "dtype": "float32",
+   "offset": 130774272,
+   "byteLength": 18432
+  },
+  {
+   "name": "transformer_blocks.4.norm_s.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 130792704,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.4.s_attn.graph_dist_scale",
+   "shape": [
+    1
+   ],
+   "dtype": "float32",
+   "offset": 130794752,
+   "byteLength": 4
+  },
+  {
+   "name": "transformer_blocks.4.s_attn.graph_rel_scale",
+   "shape": [
+    1
+   ],
+   "dtype": "float32",
+   "offset": 130795008,
+   "byteLength": 4
+  },
+  {
+   "name": "transformer_blocks.4.s_attn.rope.spectral_encoder.phi.0.weight",
+   "shape": [
+    64,
+    1
+   ],
+   "dtype": "float32",
+   "offset": 130795264,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.4.s_attn.rope.spectral_encoder.phi.0.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 130795520,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.4.s_attn.rope.spectral_encoder.phi.2.weight",
+   "shape": [
+    64,
+    64
+   ],
+   "dtype": "float32",
+   "offset": 130795776,
+   "byteLength": 16384
+  },
+  {
+   "name": "transformer_blocks.4.s_attn.rope.spectral_encoder.phi.2.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 130812160,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.4.s_attn.rope.spectral_encoder.rho.0.weight",
+   "shape": [
+    64,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 130812416,
+   "byteLength": 131072
+  },
+  {
+   "name": "transformer_blocks.4.s_attn.rope.spectral_encoder.rho.0.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 130943488,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.4.s_attn.rope.spectral_encoder.rho.2.weight",
+   "shape": [
+    32,
+    64
+   ],
+   "dtype": "float32",
+   "offset": 130943744,
+   "byteLength": 8192
+  },
+  {
+   "name": "transformer_blocks.4.s_attn.rope.spectral_encoder.rho.2.bias",
+   "shape": [
+    32
+   ],
+   "dtype": "float32",
+   "offset": 130951936,
+   "byteLength": 128
+  },
+  {
+   "name": "transformer_blocks.4.s_attn.qkv.weight",
+   "shape": [
+    1536,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 130952192,
+   "byteLength": 3145728
+  },
+  {
+   "name": "transformer_blocks.4.s_attn.qkv.bias",
+   "shape": [
+    1536
+   ],
+   "dtype": "float32",
+   "offset": 134097920,
+   "byteLength": 6144
+  },
+  {
+   "name": "transformer_blocks.4.s_attn.proj.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 134104064,
+   "byteLength": 1048576
+  },
+  {
+   "name": "transformer_blocks.4.s_attn.proj.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 135152640,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.4.s_attn.q_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 135154688,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.4.s_attn.k_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 135154944,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.4.s_attn.graph_dist_embedding.weight",
+   "shape": [
+    6,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 135155200,
+   "byteLength": 3072
+  },
+  {
+   "name": "transformer_blocks.4.s_attn.graph_dist_proj.weight",
+   "shape": [
+    8,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 135158272,
+   "byteLength": 4096
+  },
+  {
+   "name": "transformer_blocks.4.s_attn.graph_dist_proj.bias",
+   "shape": [
+    8
+   ],
+   "dtype": "float32",
+   "offset": 135162368,
+   "byteLength": 32
+  },
+  {
+   "name": "transformer_blocks.4.s_attn.graph_rel_embedding.weight",
+   "shape": [
+    6,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 135162624,
+   "byteLength": 3072
+  },
+  {
+   "name": "transformer_blocks.4.s_attn.graph_rel_proj.weight",
+   "shape": [
+    8,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 135165696,
+   "byteLength": 4096
+  },
+  {
+   "name": "transformer_blocks.4.s_attn.graph_rel_proj.bias",
+   "shape": [
+    8
+   ],
+   "dtype": "float32",
+   "offset": 135169792,
+   "byteLength": 32
+  },
+  {
+   "name": "transformer_blocks.4.norm_t.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 135170048,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.4.t_attn.qkv.weight",
+   "shape": [
+    1536,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 135172096,
+   "byteLength": 3145728
+  },
+  {
+   "name": "transformer_blocks.4.t_attn.qkv.bias",
+   "shape": [
+    1536
+   ],
+   "dtype": "float32",
+   "offset": 138317824,
+   "byteLength": 6144
+  },
+  {
+   "name": "transformer_blocks.4.t_attn.proj.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 138323968,
+   "byteLength": 1048576
+  },
+  {
+   "name": "transformer_blocks.4.t_attn.proj.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 139372544,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.4.t_attn.q_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 139374592,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.4.t_attn.k_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 139374848,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.4.norm_mlp.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 139375104,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.4.mlp.w12.weight",
+   "shape": [
+    2730,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 139377152,
+   "byteLength": 5591040
+  },
+  {
+   "name": "transformer_blocks.4.mlp.w12.bias",
+   "shape": [
+    2730
+   ],
+   "dtype": "float32",
+   "offset": 144968192,
+   "byteLength": 10920
+  },
+  {
+   "name": "transformer_blocks.4.mlp.w3.weight",
+   "shape": [
+    512,
+    1365
+   ],
+   "dtype": "float32",
+   "offset": 144979200,
+   "byteLength": 2795520
+  },
+  {
+   "name": "transformer_blocks.4.mlp.w3.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 147774720,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.4.adaLN_modulation.1.weight",
+   "shape": [
+    4608,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 147776768,
+   "byteLength": 9437184
+  },
+  {
+   "name": "transformer_blocks.4.adaLN_modulation.1.bias",
+   "shape": [
+    4608
+   ],
+   "dtype": "float32",
+   "offset": 157213952,
+   "byteLength": 18432
+  },
+  {
+   "name": "transformer_blocks.5.norm_s.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 157232384,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.5.s_attn.graph_dist_scale",
+   "shape": [
+    1
+   ],
+   "dtype": "float32",
+   "offset": 157234432,
+   "byteLength": 4
+  },
+  {
+   "name": "transformer_blocks.5.s_attn.graph_rel_scale",
+   "shape": [
+    1
+   ],
+   "dtype": "float32",
+   "offset": 157234688,
+   "byteLength": 4
+  },
+  {
+   "name": "transformer_blocks.5.s_attn.rope.spectral_encoder.phi.0.weight",
+   "shape": [
+    64,
+    1
+   ],
+   "dtype": "float32",
+   "offset": 157234944,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.5.s_attn.rope.spectral_encoder.phi.0.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 157235200,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.5.s_attn.rope.spectral_encoder.phi.2.weight",
+   "shape": [
+    64,
+    64
+   ],
+   "dtype": "float32",
+   "offset": 157235456,
+   "byteLength": 16384
+  },
+  {
+   "name": "transformer_blocks.5.s_attn.rope.spectral_encoder.phi.2.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 157251840,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.5.s_attn.rope.spectral_encoder.rho.0.weight",
+   "shape": [
+    64,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 157252096,
+   "byteLength": 131072
+  },
+  {
+   "name": "transformer_blocks.5.s_attn.rope.spectral_encoder.rho.0.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 157383168,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.5.s_attn.rope.spectral_encoder.rho.2.weight",
+   "shape": [
+    32,
+    64
+   ],
+   "dtype": "float32",
+   "offset": 157383424,
+   "byteLength": 8192
+  },
+  {
+   "name": "transformer_blocks.5.s_attn.rope.spectral_encoder.rho.2.bias",
+   "shape": [
+    32
+   ],
+   "dtype": "float32",
+   "offset": 157391616,
+   "byteLength": 128
+  },
+  {
+   "name": "transformer_blocks.5.s_attn.qkv.weight",
+   "shape": [
+    1536,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 157391872,
+   "byteLength": 3145728
+  },
+  {
+   "name": "transformer_blocks.5.s_attn.qkv.bias",
+   "shape": [
+    1536
+   ],
+   "dtype": "float32",
+   "offset": 160537600,
+   "byteLength": 6144
+  },
+  {
+   "name": "transformer_blocks.5.s_attn.proj.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 160543744,
+   "byteLength": 1048576
+  },
+  {
+   "name": "transformer_blocks.5.s_attn.proj.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 161592320,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.5.s_attn.q_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 161594368,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.5.s_attn.k_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 161594624,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.5.s_attn.graph_dist_embedding.weight",
+   "shape": [
+    6,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 161594880,
+   "byteLength": 3072
+  },
+  {
+   "name": "transformer_blocks.5.s_attn.graph_dist_proj.weight",
+   "shape": [
+    8,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 161597952,
+   "byteLength": 4096
+  },
+  {
+   "name": "transformer_blocks.5.s_attn.graph_dist_proj.bias",
+   "shape": [
+    8
+   ],
+   "dtype": "float32",
+   "offset": 161602048,
+   "byteLength": 32
+  },
+  {
+   "name": "transformer_blocks.5.s_attn.graph_rel_embedding.weight",
+   "shape": [
+    6,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 161602304,
+   "byteLength": 3072
+  },
+  {
+   "name": "transformer_blocks.5.s_attn.graph_rel_proj.weight",
+   "shape": [
+    8,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 161605376,
+   "byteLength": 4096
+  },
+  {
+   "name": "transformer_blocks.5.s_attn.graph_rel_proj.bias",
+   "shape": [
+    8
+   ],
+   "dtype": "float32",
+   "offset": 161609472,
+   "byteLength": 32
+  },
+  {
+   "name": "transformer_blocks.5.norm_t.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 161609728,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.5.t_attn.qkv.weight",
+   "shape": [
+    1536,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 161611776,
+   "byteLength": 3145728
+  },
+  {
+   "name": "transformer_blocks.5.t_attn.qkv.bias",
+   "shape": [
+    1536
+   ],
+   "dtype": "float32",
+   "offset": 164757504,
+   "byteLength": 6144
+  },
+  {
+   "name": "transformer_blocks.5.t_attn.proj.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 164763648,
+   "byteLength": 1048576
+  },
+  {
+   "name": "transformer_blocks.5.t_attn.proj.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 165812224,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.5.t_attn.q_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 165814272,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.5.t_attn.k_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 165814528,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.5.norm_mlp.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 165814784,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.5.mlp.w12.weight",
+   "shape": [
+    2730,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 165816832,
+   "byteLength": 5591040
+  },
+  {
+   "name": "transformer_blocks.5.mlp.w12.bias",
+   "shape": [
+    2730
+   ],
+   "dtype": "float32",
+   "offset": 171407872,
+   "byteLength": 10920
+  },
+  {
+   "name": "transformer_blocks.5.mlp.w3.weight",
+   "shape": [
+    512,
+    1365
+   ],
+   "dtype": "float32",
+   "offset": 171418880,
+   "byteLength": 2795520
+  },
+  {
+   "name": "transformer_blocks.5.mlp.w3.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 174214400,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.5.adaLN_modulation.1.weight",
+   "shape": [
+    4608,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 174216448,
+   "byteLength": 9437184
+  },
+  {
+   "name": "transformer_blocks.5.adaLN_modulation.1.bias",
+   "shape": [
+    4608
+   ],
+   "dtype": "float32",
+   "offset": 183653632,
+   "byteLength": 18432
+  },
+  {
+   "name": "transformer_blocks.6.norm_s.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 183672064,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.6.s_attn.graph_dist_scale",
+   "shape": [
+    1
+   ],
+   "dtype": "float32",
+   "offset": 183674112,
+   "byteLength": 4
+  },
+  {
+   "name": "transformer_blocks.6.s_attn.graph_rel_scale",
+   "shape": [
+    1
+   ],
+   "dtype": "float32",
+   "offset": 183674368,
+   "byteLength": 4
+  },
+  {
+   "name": "transformer_blocks.6.s_attn.rope.spectral_encoder.phi.0.weight",
+   "shape": [
+    64,
+    1
+   ],
+   "dtype": "float32",
+   "offset": 183674624,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.6.s_attn.rope.spectral_encoder.phi.0.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 183674880,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.6.s_attn.rope.spectral_encoder.phi.2.weight",
+   "shape": [
+    64,
+    64
+   ],
+   "dtype": "float32",
+   "offset": 183675136,
+   "byteLength": 16384
+  },
+  {
+   "name": "transformer_blocks.6.s_attn.rope.spectral_encoder.phi.2.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 183691520,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.6.s_attn.rope.spectral_encoder.rho.0.weight",
+   "shape": [
+    64,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 183691776,
+   "byteLength": 131072
+  },
+  {
+   "name": "transformer_blocks.6.s_attn.rope.spectral_encoder.rho.0.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 183822848,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.6.s_attn.rope.spectral_encoder.rho.2.weight",
+   "shape": [
+    32,
+    64
+   ],
+   "dtype": "float32",
+   "offset": 183823104,
+   "byteLength": 8192
+  },
+  {
+   "name": "transformer_blocks.6.s_attn.rope.spectral_encoder.rho.2.bias",
+   "shape": [
+    32
+   ],
+   "dtype": "float32",
+   "offset": 183831296,
+   "byteLength": 128
+  },
+  {
+   "name": "transformer_blocks.6.s_attn.qkv.weight",
+   "shape": [
+    1536,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 183831552,
+   "byteLength": 3145728
+  },
+  {
+   "name": "transformer_blocks.6.s_attn.qkv.bias",
+   "shape": [
+    1536
+   ],
+   "dtype": "float32",
+   "offset": 186977280,
+   "byteLength": 6144
+  },
+  {
+   "name": "transformer_blocks.6.s_attn.proj.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 186983424,
+   "byteLength": 1048576
+  },
+  {
+   "name": "transformer_blocks.6.s_attn.proj.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 188032000,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.6.s_attn.q_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 188034048,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.6.s_attn.k_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 188034304,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.6.s_attn.graph_dist_embedding.weight",
+   "shape": [
+    6,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 188034560,
+   "byteLength": 3072
+  },
+  {
+   "name": "transformer_blocks.6.s_attn.graph_dist_proj.weight",
+   "shape": [
+    8,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 188037632,
+   "byteLength": 4096
+  },
+  {
+   "name": "transformer_blocks.6.s_attn.graph_dist_proj.bias",
+   "shape": [
+    8
+   ],
+   "dtype": "float32",
+   "offset": 188041728,
+   "byteLength": 32
+  },
+  {
+   "name": "transformer_blocks.6.s_attn.graph_rel_embedding.weight",
+   "shape": [
+    6,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 188041984,
+   "byteLength": 3072
+  },
+  {
+   "name": "transformer_blocks.6.s_attn.graph_rel_proj.weight",
+   "shape": [
+    8,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 188045056,
+   "byteLength": 4096
+  },
+  {
+   "name": "transformer_blocks.6.s_attn.graph_rel_proj.bias",
+   "shape": [
+    8
+   ],
+   "dtype": "float32",
+   "offset": 188049152,
+   "byteLength": 32
+  },
+  {
+   "name": "transformer_blocks.6.norm_t.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 188049408,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.6.t_attn.qkv.weight",
+   "shape": [
+    1536,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 188051456,
+   "byteLength": 3145728
+  },
+  {
+   "name": "transformer_blocks.6.t_attn.qkv.bias",
+   "shape": [
+    1536
+   ],
+   "dtype": "float32",
+   "offset": 191197184,
+   "byteLength": 6144
+  },
+  {
+   "name": "transformer_blocks.6.t_attn.proj.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 191203328,
+   "byteLength": 1048576
+  },
+  {
+   "name": "transformer_blocks.6.t_attn.proj.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 192251904,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.6.t_attn.q_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 192253952,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.6.t_attn.k_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 192254208,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.6.norm_mlp.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 192254464,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.6.mlp.w12.weight",
+   "shape": [
+    2730,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 192256512,
+   "byteLength": 5591040
+  },
+  {
+   "name": "transformer_blocks.6.mlp.w12.bias",
+   "shape": [
+    2730
+   ],
+   "dtype": "float32",
+   "offset": 197847552,
+   "byteLength": 10920
+  },
+  {
+   "name": "transformer_blocks.6.mlp.w3.weight",
+   "shape": [
+    512,
+    1365
+   ],
+   "dtype": "float32",
+   "offset": 197858560,
+   "byteLength": 2795520
+  },
+  {
+   "name": "transformer_blocks.6.mlp.w3.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 200654080,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.6.adaLN_modulation.1.weight",
+   "shape": [
+    4608,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 200656128,
+   "byteLength": 9437184
+  },
+  {
+   "name": "transformer_blocks.6.adaLN_modulation.1.bias",
+   "shape": [
+    4608
+   ],
+   "dtype": "float32",
+   "offset": 210093312,
+   "byteLength": 18432
+  },
+  {
+   "name": "transformer_blocks.7.norm_s.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 210111744,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.7.s_attn.graph_dist_scale",
+   "shape": [
+    1
+   ],
+   "dtype": "float32",
+   "offset": 210113792,
+   "byteLength": 4
+  },
+  {
+   "name": "transformer_blocks.7.s_attn.graph_rel_scale",
+   "shape": [
+    1
+   ],
+   "dtype": "float32",
+   "offset": 210114048,
+   "byteLength": 4
+  },
+  {
+   "name": "transformer_blocks.7.s_attn.rope.spectral_encoder.phi.0.weight",
+   "shape": [
+    64,
+    1
+   ],
+   "dtype": "float32",
+   "offset": 210114304,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.7.s_attn.rope.spectral_encoder.phi.0.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 210114560,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.7.s_attn.rope.spectral_encoder.phi.2.weight",
+   "shape": [
+    64,
+    64
+   ],
+   "dtype": "float32",
+   "offset": 210114816,
+   "byteLength": 16384
+  },
+  {
+   "name": "transformer_blocks.7.s_attn.rope.spectral_encoder.phi.2.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 210131200,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.7.s_attn.rope.spectral_encoder.rho.0.weight",
+   "shape": [
+    64,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 210131456,
+   "byteLength": 131072
+  },
+  {
+   "name": "transformer_blocks.7.s_attn.rope.spectral_encoder.rho.0.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 210262528,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.7.s_attn.rope.spectral_encoder.rho.2.weight",
+   "shape": [
+    32,
+    64
+   ],
+   "dtype": "float32",
+   "offset": 210262784,
+   "byteLength": 8192
+  },
+  {
+   "name": "transformer_blocks.7.s_attn.rope.spectral_encoder.rho.2.bias",
+   "shape": [
+    32
+   ],
+   "dtype": "float32",
+   "offset": 210270976,
+   "byteLength": 128
+  },
+  {
+   "name": "transformer_blocks.7.s_attn.qkv.weight",
+   "shape": [
+    1536,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 210271232,
+   "byteLength": 3145728
+  },
+  {
+   "name": "transformer_blocks.7.s_attn.qkv.bias",
+   "shape": [
+    1536
+   ],
+   "dtype": "float32",
+   "offset": 213416960,
+   "byteLength": 6144
+  },
+  {
+   "name": "transformer_blocks.7.s_attn.proj.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 213423104,
+   "byteLength": 1048576
+  },
+  {
+   "name": "transformer_blocks.7.s_attn.proj.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 214471680,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.7.s_attn.q_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 214473728,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.7.s_attn.k_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 214473984,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.7.s_attn.graph_dist_embedding.weight",
+   "shape": [
+    6,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 214474240,
+   "byteLength": 3072
+  },
+  {
+   "name": "transformer_blocks.7.s_attn.graph_dist_proj.weight",
+   "shape": [
+    8,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 214477312,
+   "byteLength": 4096
+  },
+  {
+   "name": "transformer_blocks.7.s_attn.graph_dist_proj.bias",
+   "shape": [
+    8
+   ],
+   "dtype": "float32",
+   "offset": 214481408,
+   "byteLength": 32
+  },
+  {
+   "name": "transformer_blocks.7.s_attn.graph_rel_embedding.weight",
+   "shape": [
+    6,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 214481664,
+   "byteLength": 3072
+  },
+  {
+   "name": "transformer_blocks.7.s_attn.graph_rel_proj.weight",
+   "shape": [
+    8,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 214484736,
+   "byteLength": 4096
+  },
+  {
+   "name": "transformer_blocks.7.s_attn.graph_rel_proj.bias",
+   "shape": [
+    8
+   ],
+   "dtype": "float32",
+   "offset": 214488832,
+   "byteLength": 32
+  },
+  {
+   "name": "transformer_blocks.7.norm_t.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 214489088,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.7.t_attn.qkv.weight",
+   "shape": [
+    1536,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 214491136,
+   "byteLength": 3145728
+  },
+  {
+   "name": "transformer_blocks.7.t_attn.qkv.bias",
+   "shape": [
+    1536
+   ],
+   "dtype": "float32",
+   "offset": 217636864,
+   "byteLength": 6144
+  },
+  {
+   "name": "transformer_blocks.7.t_attn.proj.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 217643008,
+   "byteLength": 1048576
+  },
+  {
+   "name": "transformer_blocks.7.t_attn.proj.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 218691584,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.7.t_attn.q_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 218693632,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.7.t_attn.k_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 218693888,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.7.norm_mlp.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 218694144,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.7.mlp.w12.weight",
+   "shape": [
+    2730,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 218696192,
+   "byteLength": 5591040
+  },
+  {
+   "name": "transformer_blocks.7.mlp.w12.bias",
+   "shape": [
+    2730
+   ],
+   "dtype": "float32",
+   "offset": 224287232,
+   "byteLength": 10920
+  },
+  {
+   "name": "transformer_blocks.7.mlp.w3.weight",
+   "shape": [
+    512,
+    1365
+   ],
+   "dtype": "float32",
+   "offset": 224298240,
+   "byteLength": 2795520
+  },
+  {
+   "name": "transformer_blocks.7.mlp.w3.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 227093760,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.7.adaLN_modulation.1.weight",
+   "shape": [
+    4608,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 227095808,
+   "byteLength": 9437184
+  },
+  {
+   "name": "transformer_blocks.7.adaLN_modulation.1.bias",
+   "shape": [
+    4608
+   ],
+   "dtype": "float32",
+   "offset": 236532992,
+   "byteLength": 18432
+  },
+  {
+   "name": "transformer_blocks.8.norm_s.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 236551424,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.8.s_attn.graph_dist_scale",
+   "shape": [
+    1
+   ],
+   "dtype": "float32",
+   "offset": 236553472,
+   "byteLength": 4
+  },
+  {
+   "name": "transformer_blocks.8.s_attn.graph_rel_scale",
+   "shape": [
+    1
+   ],
+   "dtype": "float32",
+   "offset": 236553728,
+   "byteLength": 4
+  },
+  {
+   "name": "transformer_blocks.8.s_attn.rope.spectral_encoder.phi.0.weight",
+   "shape": [
+    64,
+    1
+   ],
+   "dtype": "float32",
+   "offset": 236553984,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.8.s_attn.rope.spectral_encoder.phi.0.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 236554240,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.8.s_attn.rope.spectral_encoder.phi.2.weight",
+   "shape": [
+    64,
+    64
+   ],
+   "dtype": "float32",
+   "offset": 236554496,
+   "byteLength": 16384
+  },
+  {
+   "name": "transformer_blocks.8.s_attn.rope.spectral_encoder.phi.2.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 236570880,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.8.s_attn.rope.spectral_encoder.rho.0.weight",
+   "shape": [
+    64,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 236571136,
+   "byteLength": 131072
+  },
+  {
+   "name": "transformer_blocks.8.s_attn.rope.spectral_encoder.rho.0.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 236702208,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.8.s_attn.rope.spectral_encoder.rho.2.weight",
+   "shape": [
+    32,
+    64
+   ],
+   "dtype": "float32",
+   "offset": 236702464,
+   "byteLength": 8192
+  },
+  {
+   "name": "transformer_blocks.8.s_attn.rope.spectral_encoder.rho.2.bias",
+   "shape": [
+    32
+   ],
+   "dtype": "float32",
+   "offset": 236710656,
+   "byteLength": 128
+  },
+  {
+   "name": "transformer_blocks.8.s_attn.qkv.weight",
+   "shape": [
+    1536,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 236710912,
+   "byteLength": 3145728
+  },
+  {
+   "name": "transformer_blocks.8.s_attn.qkv.bias",
+   "shape": [
+    1536
+   ],
+   "dtype": "float32",
+   "offset": 239856640,
+   "byteLength": 6144
+  },
+  {
+   "name": "transformer_blocks.8.s_attn.proj.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 239862784,
+   "byteLength": 1048576
+  },
+  {
+   "name": "transformer_blocks.8.s_attn.proj.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 240911360,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.8.s_attn.q_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 240913408,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.8.s_attn.k_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 240913664,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.8.s_attn.graph_dist_embedding.weight",
+   "shape": [
+    6,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 240913920,
+   "byteLength": 3072
+  },
+  {
+   "name": "transformer_blocks.8.s_attn.graph_dist_proj.weight",
+   "shape": [
+    8,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 240916992,
+   "byteLength": 4096
+  },
+  {
+   "name": "transformer_blocks.8.s_attn.graph_dist_proj.bias",
+   "shape": [
+    8
+   ],
+   "dtype": "float32",
+   "offset": 240921088,
+   "byteLength": 32
+  },
+  {
+   "name": "transformer_blocks.8.s_attn.graph_rel_embedding.weight",
+   "shape": [
+    6,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 240921344,
+   "byteLength": 3072
+  },
+  {
+   "name": "transformer_blocks.8.s_attn.graph_rel_proj.weight",
+   "shape": [
+    8,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 240924416,
+   "byteLength": 4096
+  },
+  {
+   "name": "transformer_blocks.8.s_attn.graph_rel_proj.bias",
+   "shape": [
+    8
+   ],
+   "dtype": "float32",
+   "offset": 240928512,
+   "byteLength": 32
+  },
+  {
+   "name": "transformer_blocks.8.norm_t.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 240928768,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.8.t_attn.qkv.weight",
+   "shape": [
+    1536,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 240930816,
+   "byteLength": 3145728
+  },
+  {
+   "name": "transformer_blocks.8.t_attn.qkv.bias",
+   "shape": [
+    1536
+   ],
+   "dtype": "float32",
+   "offset": 244076544,
+   "byteLength": 6144
+  },
+  {
+   "name": "transformer_blocks.8.t_attn.proj.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 244082688,
+   "byteLength": 1048576
+  },
+  {
+   "name": "transformer_blocks.8.t_attn.proj.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 245131264,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.8.t_attn.q_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 245133312,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.8.t_attn.k_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 245133568,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.8.norm_mlp.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 245133824,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.8.mlp.w12.weight",
+   "shape": [
+    2730,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 245135872,
+   "byteLength": 5591040
+  },
+  {
+   "name": "transformer_blocks.8.mlp.w12.bias",
+   "shape": [
+    2730
+   ],
+   "dtype": "float32",
+   "offset": 250726912,
+   "byteLength": 10920
+  },
+  {
+   "name": "transformer_blocks.8.mlp.w3.weight",
+   "shape": [
+    512,
+    1365
+   ],
+   "dtype": "float32",
+   "offset": 250737920,
+   "byteLength": 2795520
+  },
+  {
+   "name": "transformer_blocks.8.mlp.w3.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 253533440,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.8.adaLN_modulation.1.weight",
+   "shape": [
+    4608,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 253535488,
+   "byteLength": 9437184
+  },
+  {
+   "name": "transformer_blocks.8.adaLN_modulation.1.bias",
+   "shape": [
+    4608
+   ],
+   "dtype": "float32",
+   "offset": 262972672,
+   "byteLength": 18432
+  },
+  {
+   "name": "transformer_blocks.9.norm_s.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 262991104,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.9.s_attn.graph_dist_scale",
+   "shape": [
+    1
+   ],
+   "dtype": "float32",
+   "offset": 262993152,
+   "byteLength": 4
+  },
+  {
+   "name": "transformer_blocks.9.s_attn.graph_rel_scale",
+   "shape": [
+    1
+   ],
+   "dtype": "float32",
+   "offset": 262993408,
+   "byteLength": 4
+  },
+  {
+   "name": "transformer_blocks.9.s_attn.rope.spectral_encoder.phi.0.weight",
+   "shape": [
+    64,
+    1
+   ],
+   "dtype": "float32",
+   "offset": 262993664,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.9.s_attn.rope.spectral_encoder.phi.0.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 262993920,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.9.s_attn.rope.spectral_encoder.phi.2.weight",
+   "shape": [
+    64,
+    64
+   ],
+   "dtype": "float32",
+   "offset": 262994176,
+   "byteLength": 16384
+  },
+  {
+   "name": "transformer_blocks.9.s_attn.rope.spectral_encoder.phi.2.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 263010560,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.9.s_attn.rope.spectral_encoder.rho.0.weight",
+   "shape": [
+    64,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 263010816,
+   "byteLength": 131072
+  },
+  {
+   "name": "transformer_blocks.9.s_attn.rope.spectral_encoder.rho.0.bias",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 263141888,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.9.s_attn.rope.spectral_encoder.rho.2.weight",
+   "shape": [
+    32,
+    64
+   ],
+   "dtype": "float32",
+   "offset": 263142144,
+   "byteLength": 8192
+  },
+  {
+   "name": "transformer_blocks.9.s_attn.rope.spectral_encoder.rho.2.bias",
+   "shape": [
+    32
+   ],
+   "dtype": "float32",
+   "offset": 263150336,
+   "byteLength": 128
+  },
+  {
+   "name": "transformer_blocks.9.s_attn.qkv.weight",
+   "shape": [
+    1536,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 263150592,
+   "byteLength": 3145728
+  },
+  {
+   "name": "transformer_blocks.9.s_attn.qkv.bias",
+   "shape": [
+    1536
+   ],
+   "dtype": "float32",
+   "offset": 266296320,
+   "byteLength": 6144
+  },
+  {
+   "name": "transformer_blocks.9.s_attn.proj.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 266302464,
+   "byteLength": 1048576
+  },
+  {
+   "name": "transformer_blocks.9.s_attn.proj.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 267351040,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.9.s_attn.q_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 267353088,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.9.s_attn.k_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 267353344,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.9.s_attn.graph_dist_embedding.weight",
+   "shape": [
+    6,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 267353600,
+   "byteLength": 3072
+  },
+  {
+   "name": "transformer_blocks.9.s_attn.graph_dist_proj.weight",
+   "shape": [
+    8,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 267356672,
+   "byteLength": 4096
+  },
+  {
+   "name": "transformer_blocks.9.s_attn.graph_dist_proj.bias",
+   "shape": [
+    8
+   ],
+   "dtype": "float32",
+   "offset": 267360768,
+   "byteLength": 32
+  },
+  {
+   "name": "transformer_blocks.9.s_attn.graph_rel_embedding.weight",
+   "shape": [
+    6,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 267361024,
+   "byteLength": 3072
+  },
+  {
+   "name": "transformer_blocks.9.s_attn.graph_rel_proj.weight",
+   "shape": [
+    8,
+    128
+   ],
+   "dtype": "float32",
+   "offset": 267364096,
+   "byteLength": 4096
+  },
+  {
+   "name": "transformer_blocks.9.s_attn.graph_rel_proj.bias",
+   "shape": [
+    8
+   ],
+   "dtype": "float32",
+   "offset": 267368192,
+   "byteLength": 32
+  },
+  {
+   "name": "transformer_blocks.9.norm_t.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 267368448,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.9.t_attn.qkv.weight",
+   "shape": [
+    1536,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 267370496,
+   "byteLength": 3145728
+  },
+  {
+   "name": "transformer_blocks.9.t_attn.qkv.bias",
+   "shape": [
+    1536
+   ],
+   "dtype": "float32",
+   "offset": 270516224,
+   "byteLength": 6144
+  },
+  {
+   "name": "transformer_blocks.9.t_attn.proj.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 270522368,
+   "byteLength": 1048576
+  },
+  {
+   "name": "transformer_blocks.9.t_attn.proj.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 271570944,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.9.t_attn.q_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 271572992,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.9.t_attn.k_norm.weight",
+   "shape": [
+    64
+   ],
+   "dtype": "float32",
+   "offset": 271573248,
+   "byteLength": 256
+  },
+  {
+   "name": "transformer_blocks.9.norm_mlp.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 271573504,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.9.mlp.w12.weight",
+   "shape": [
+    2730,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 271575552,
+   "byteLength": 5591040
+  },
+  {
+   "name": "transformer_blocks.9.mlp.w12.bias",
+   "shape": [
+    2730
+   ],
+   "dtype": "float32",
+   "offset": 277166592,
+   "byteLength": 10920
+  },
+  {
+   "name": "transformer_blocks.9.mlp.w3.weight",
+   "shape": [
+    512,
+    1365
+   ],
+   "dtype": "float32",
+   "offset": 277177600,
+   "byteLength": 2795520
+  },
+  {
+   "name": "transformer_blocks.9.mlp.w3.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 279973120,
+   "byteLength": 2048
+  },
+  {
+   "name": "transformer_blocks.9.adaLN_modulation.1.weight",
+   "shape": [
+    4608,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 279975168,
+   "byteLength": 9437184
+  },
+  {
+   "name": "transformer_blocks.9.adaLN_modulation.1.bias",
+   "shape": [
+    4608
+   ],
+   "dtype": "float32",
+   "offset": 289412352,
+   "byteLength": 18432
+  },
+  {
+   "name": "final_layer.norm_final.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 289430784,
+   "byteLength": 2048
+  },
+  {
+   "name": "final_layer.root_cross_attn.in_proj_weight",
+   "shape": [
+    1536,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 289432832,
+   "byteLength": 3145728
+  },
+  {
+   "name": "final_layer.root_cross_attn.in_proj_bias",
+   "shape": [
+    1536
+   ],
+   "dtype": "float32",
+   "offset": 292578560,
+   "byteLength": 6144
+  },
+  {
+   "name": "final_layer.root_cross_attn.out_proj.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 292584704,
+   "byteLength": 1048576
+  },
+  {
+   "name": "final_layer.root_cross_attn.out_proj.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 293633280,
+   "byteLength": 2048
+  },
+  {
+   "name": "final_layer.root_cross_norm_q.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 293635328,
+   "byteLength": 2048
+  },
+  {
+   "name": "final_layer.root_cross_norm_kv.weight",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 293637376,
+   "byteLength": 2048
+  },
+  {
+   "name": "final_layer.root_out.0.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 293639424,
+   "byteLength": 1048576
+  },
+  {
+   "name": "final_layer.root_out.0.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 294688000,
+   "byteLength": 2048
+  },
+  {
+   "name": "final_layer.root_out.2.weight",
+   "shape": [
+    12,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 294690048,
+   "byteLength": 24576
+  },
+  {
+   "name": "final_layer.root_out.2.bias",
+   "shape": [
+    12
+   ],
+   "dtype": "float32",
+   "offset": 294714624,
+   "byteLength": 48
+  },
+  {
+   "name": "final_layer.joint_out.0.weight",
+   "shape": [
+    512,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 294714880,
+   "byteLength": 1048576
+  },
+  {
+   "name": "final_layer.joint_out.0.bias",
+   "shape": [
+    512
+   ],
+   "dtype": "float32",
+   "offset": 295763456,
+   "byteLength": 2048
+  },
+  {
+   "name": "final_layer.joint_out.2.weight",
+   "shape": [
+    12,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 295765504,
+   "byteLength": 24576
+  },
+  {
+   "name": "final_layer.joint_out.2.bias",
+   "shape": [
+    12
+   ],
+   "dtype": "float32",
+   "offset": 295790080,
+   "byteLength": 48
+  },
+  {
+   "name": "final_layer.adaLN_modulation.1.weight",
+   "shape": [
+    1024,
+    512
+   ],
+   "dtype": "float32",
+   "offset": 295790336,
+   "byteLength": 2097152
+  },
+  {
+   "name": "final_layer.adaLN_modulation.1.bias",
+   "shape": [
+    1024
+   ],
+   "dtype": "float32",
+   "offset": 297887488,
+   "byteLength": 4096
+  }
+ ],
+ "verification": {
+  "capture": "PyTorch forward capture: unitree_go2, two prompts, seed 10",
+  "bitwise_equal": true,
+  "cond_max_abs_diff": 0.0,
+  "uncond_max_abs_diff": 0.0
+ }
+}

unimate/unimate-denoiser-fp32.bin

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