Add v8 browser files

.gitattributes

@@ -36,3 +36,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 webgpu/post_joint_100k/weights.f32 filter=lfs diff=lfs merge=lfs -text
 webgpu/ck_135000/weights.f32 filter=lfs diff=lfs merge=lfs -text
 webgpu/ck_240000/weights.f32 filter=lfs diff=lfs merge=lfs -text
+webgpu/v8_ck_240000/weights.f32 filter=lfs diff=lfs merge=lfs -text

webgpu/v8_ck_240000/manifest.json

@@ -0,0 +1,3030 @@
+{
+  "format": "wsfmdock_webgpu_v8",
+  "method": "continuous_docking_backbone_segments_alldepth_secant_x1_ck_v8",
+  "stage": "ck",
+  "iteration": 240000,
+  "checkpoint_sha256": "425083dee00c40a570dbf5495aeeeb1503b6d3f4c32516c1e1edcdce203e77ac",
+  "weight_file": "weights.f32",
+  "weight_bytes": 331541488,
+  "weight_sha256": "ec3998a6b4013fede23f61cbd83cb224bb86ceff612a13bcb2f3bd8241910227",
+  "weight_precision": "float32",
+  "activation_precision": "float32",
+  "accumulation_precision": "float32",
+  "config": {
+    "dim": 408,
+    "depth": 12,
+    "heads": 12,
+    "scalar_head_dim": 34,
+    "query_points": 6,
+    "point_values": 10,
+    "ff_hidden_dim": 2040,
+    "rope_base": 1000.0,
+    "flex_block_size": 128,
+    "a_e": 1.0,
+    "endpoint_update_layers": 3,
+    "endpoint_update_stride": 2,
+    "ck_suffix_layers": 5,
+    "multiscale_spatial_heads": 6,
+    "spatial_scale_min": 0.001,
+    "spatial_scale_max": 0.3,
+    "bond_bias_init_min": 0.0,
+    "bond_bias_init_max": 8.0
+  },
+  "embedding_offsets": {
+    "atomic": 0,
+    "role": 128,
+    "residue": 133,
+    "atom_name": 154
+  },
+  "endpoint_block_indices": [
+    7,
+    9,
+    11
+  ],
+  "finite_start_block": 7,
+  "lateral_width": 1024,
+  "sampling": {
+    "initial_molecule_scale": 10.0,
+    "process_molecule_scale": 1.0,
+    "sidechain_scale": 0.5
+  },
+  "time_frequencies": [
+    6.641670227050781,
+    9.074256896972656,
+    -1.7723369598388672,
+    8.445029258728027,
+    -1.5641759634017944,
+    -2.2122936248779297,
+    3.528808355331421,
+    8.99981689453125,
+    0.9290838241577148,
+    9.222551345825195,
+    -10.863503456115723,
+    2.4348864555358887,
+    -2.269798517227173,
+    -0.2420644760131836,
+    2.674729347229004,
+    -3.7042477130889893,
+    -5.670783519744873,
+    -5.274287700653076,
+    -7.8033037185668945,
+    -5.01579475402832,
+    7.434439659118652,
+    -12.461644172668457,
+    2.83498215675354,
+    -7.307047367095947,
+    4.0082855224609375,
+    5.530520439147949,
+    3.0829312801361084,
+    -8.378491401672363,
+    -0.04035911709070206,
+    -16.81785011291504,
+    -13.761927604675293,
+    -2.997864246368408,
+    16.05389976501465,
+    3.4412286281585693,
+    3.411165475845337,
+    -3.1478991508483887,
+    4.809675216674805,
+    0.025110600516200066,
+    -1.943206787109375,
+    5.45672082901001,
+    -3.1018519401550293,
+    -1.0861843824386597,
+    0.3605387806892395,
+    -2.801668167114258,
+    -1.5190898180007935,
+    7.443546772003174,
+    -6.148499965667725,
+    4.634922981262207,
+    -5.421417236328125,
+    -3.002305269241333,
+    0.2525463402271271,
+    -5.6707963943481445,
+    10.03635311126709,
+    -2.2733154296875,
+    4.1215033531188965,
+    6.777116298675537,
+    -10.28072452545166,
+    -14.23559856414795,
+    -8.285950660705566,
+    7.015595436096191,
+    0.049032434821128845,
+    9.946003913879395,
+    1.9397789239883423,
+    5.998835563659668,
+    3.385164976119995,
+    4.276932716369629,
+    0.6565613150596619,
+    -5.322626113891602,
+    -2.449033498764038,
+    -6.313889980316162,
+    -5.7336859703063965,
+    -0.9863746762275696,
+    -10.863755226135254,
+    10.495851516723633,
+    4.975937366485596,
+    4.330070972442627,
+    -2.076622724533081,
+    -11.964531898498535,
+    -0.2836264967918396,
+    -2.0674657821655273,
+    6.978450298309326,
+    -5.835918426513672,
+    -10.280917167663574,
+    -9.381731033325195,
+    1.9929068088531494,
+    3.750739574432373,
+    -7.248497486114502,
+    1.9181675910949707,
+    -4.356318473815918,
+    -1.9954235553741455,
+    -2.195485830307007,
+    2.0695762634277344,
+    2.8948001861572266,
+    4.890586853027344,
+    0.6092849969863892,
+    0.4583764374256134,
+    4.810478687286377,
+    -1.0248830318450928,
+    12.28849983215332,
+    14.20398235321045,
+    -6.366440773010254,
+    0.09225060790777206,
+    6.816547870635986,
+    -6.22482442855835,
+    2.7981014251708984,
+    -7.410264492034912,
+    -2.3002045154571533,
+    2.451075792312622,
+    -3.305241346359253,
+    -11.133681297302246,
+    1.9476172924041748,
+    10.692784309387207,
+    -5.845837593078613,
+    7.771844863891602,
+    -7.339365005493164,
+    1.3129369020462036,
+    -0.458981454372406,
+    7.52816915512085,
+    -0.9085550308227539,
+    1.64313805103302,
+    -10.480955123901367,
+    -4.463629245758057,
+    -4.632670879364014,
+    4.050179958343506,
+    1.8981401920318604,
+    -9.747804641723633,
+    -6.812469482421875,
+    5.181924819946289,
+    7.109494209289551,
+    -6.311796188354492,
+    -9.025218963623047,
+    -9.584630012512207,
+    1.1858493089675903,
+    1.9693129062652588,
+    3.937795400619507,
+    6.127451419830322,
+    5.037847518920898,
+    1.7954579591751099,
+    1.383825659751892,
+    -7.311039447784424,
+    -7.823827266693115,
+    5.05324125289917,
+    -0.17968334257602692,
+    -0.7940857410430908,
+    -11.556039810180664,
+    1.8041058778762817,
+    4.015260219573975,
+    -10.915335655212402,
+    9.889113426208496,
+    -12.448019981384277,
+    10.373078346252441,
+    -2.8326339721679688,
+    6.589669227600098,
+    1.6592177152633667,
+    3.208179235458374,
+    -12.19528579711914,
+    10.296189308166504,
+    -0.985230565071106,
+    -6.510705947875977,
+    -5.282708644866943,
+    -4.226221084594727,
+    2.847604990005493,
+    -0.8933730721473694,
+    3.287778854370117,
+    2.681788921356201,
+    -4.823451519012451,
+    1.890516996383667,
+    3.906057834625244,
+    -5.342893600463867,
+    7.500852584838867,
+    -1.989865779876709,
+    0.18692834675312042,
+    9.360791206359863,
+    4.973384857177734,
+    1.7561607360839844,
+    4.4882073402404785,
+    -2.5046234130859375,
+    -10.136517524719238,
+    0.6101129055023193,
+    -10.491379737854004,
+    -7.367652416229248,
+    -6.716552257537842,
+    7.242635726928711,
+    0.3392133414745331,
+    0.29238176345825195,
+    1.362377643585205,
+    3.4095120429992676,
+    2.384448766708374,
+    -6.9569597244262695,
+    -2.986891746520996,
+    -0.8865398168563843,
+    6.151259899139404,
+    -0.2196812927722931,
+    4.041287899017334,
+    -5.139440059661865,
+    -2.2782866954803467,
+    5.043351650238037,
+    -7.025182247161865,
+    -3.1190876960754395,
+    1.771507978439331,
+    -6.2102274894714355,
+    -13.74133014678955,
+    3.062546730041504,
+    -5.8471527099609375,
+    2.936671495437622,
+    -0.010395261459052563,
+    -3.1961071491241455,
+    -2.1623284816741943,
+    -10.078163146972656,
+    11.058752059936523,
+    -4.167074680328369,
+    10.563041687011719,
+    -3.9864449501037598,
+    -7.221310615539551,
+    -3.888606548309326,
+    -0.8271933794021606,
+    -2.1393656730651855,
+    0.6471344232559204,
+    3.5074894428253174,
+    -5.288480281829834,
+    -8.643198013305664,
+    8.185090065002441,
+    1.3039296865463257,
+    7.232913494110107,
+    -3.44378662109375,
+    9.343338966369629,
+    -2.0342824459075928,
+    -2.2599542140960693,
+    -5.2418718338012695,
+    0.3487952947616577,
+    3.8703646659851074,
+    6.042012691497803,
+    0.42278772592544556,
+    13.869487762451172,
+    -0.5022817254066467,
+    -7.221807956695557,
+    -4.22135591506958,
+    -0.04947290197014809,
+    -6.852050304412842,
+    1.1465415954589844,
+    5.100799083709717,
+    -3.3802196979522705,
+    0.805705726146698,
+    1.716975450515747,
+    7.655481338500977,
+    -6.982658863067627,
+    -6.420623779296875,
+    5.423731327056885,
+    2.8081438541412354,
+    -5.3651442527771,
+    8.084063529968262,
+    -1.7328189611434937,
+    4.681977272033691,
+    2.6811232566833496,
+    1.7920328378677368,
+    -2.787888526916504,
+    5.1082329750061035,
+    5.991892337799072,
+    -10.368412971496582,
+    -0.5756718516349792,
+    6.515495300292969,
+    -0.12538591027259827,
+    -4.157395839691162,
+    -12.140035629272461,
+    -1.642047643661499,
+    -4.086763381958008,
+    -6.381872653961182,
+    4.3873677253723145,
+    -8.15517520904541,
+    -3.87650203704834,
+    7.007783889770508,
+    10.363239288330078,
+    3.798348903656006,
+    10.365145683288574,
+    0.024037016555666924,
+    3.2968263626098633,
+    6.487846851348877,
+    -10.162728309631348,
+    2.0858917236328125,
+    5.433018684387207,
+    -2.282616376876831,
+    -1.1154744625091553,
+    -1.1968388557434082,
+    4.653929233551025,
+    -2.284942865371704,
+    8.478745460510254,
+    -6.246888160705566,
+    3.3941824436187744,
+    12.65295124053955,
+    8.26505184173584,
+    6.129899978637695,
+    -2.5109100341796875,
+    3.520368814468384,
+    3.898855447769165,
+    1.6922954320907593,
+    -10.867064476013184,
+    5.325526237487793,
+    7.891923904418945,
+    7.614623069763184,
+    5.974137783050537,
+    -7.50972318649292,
+    -0.6342536807060242,
+    5.877256870269775,
+    5.580990314483643,
+    3.6853978633880615,
+    3.239553213119507,
+    -7.593712329864502,
+    -0.14854973554611206,
+    11.184463500976562,
+    -0.2653779983520508,
+    9.953760147094727,
+    6.936809062957764,
+    5.194331169128418,
+    0.8417597413063049,
+    -14.449994087219238,
+    -12.2345609664917,
+    12.195137977600098,
+    -0.14118555188179016,
+    8.022275924682617,
+    1.4738085269927979,
+    11.40345287322998,
+    0.04031062871217728,
+    -0.29887816309928894,
+    1.7035884857177734,
+    0.5646899342536926,
+    4.907189846038818,
+    0.25372785329818726,
+    7.314148902893066,
+    -10.725066184997559,
+    0.6059377193450928,
+    -0.192482590675354,
+    5.410050868988037,
+    1.4159018993377686,
+    -1.1385618448257446,
+    -13.785346031188965,
+    2.5464484691619873,
+    -5.514156818389893,
+    -2.7445292472839355,
+    -8.180571556091309,
+    3.836935520172119,
+    0.9657350182533264,
+    -4.5014166831970215,
+    4.0045623779296875,
+    3.72245717048645,
+    2.7523844242095947,
+    -3.9829845428466797,
+    9.744963645935059,
+    -8.313566207885742,
+    9.44027042388916,
+    0.37592750787734985,
+    13.873886108398438,
+    -8.707624435424805,
+    2.9821553230285645,
+    4.55060338973999,
+    -10.130577087402344,
+    3.8860037326812744,
+    3.5626513957977295,
+    -7.1520094871521,
+    4.867298126220703,
+    -2.6542603969573975,
+    6.118062496185303,
+    2.9122543334960938,
+    5.623640537261963,
+    -5.836672306060791,
+    0.06568397581577301,
+    -3.3156301975250244,
+    -6.804542064666748,
+    14.254461288452148,
+    3.3197689056396484,
+    3.9057185649871826,
+    -0.12135185301303864,
+    9.842053413391113,
+    4.64711856842041,
+    0.5384713411331177,
+    -14.48873233795166,
+    -4.711231231689453,
+    7.007510662078857,
+    -1.689589262008667,
+    2.5211095809936523,
+    -0.02304122969508171,
+    8.87883186340332,
+    -4.746188640594482,
+    -6.8456597328186035,
+    -13.182085990905762,
+    0.8862783312797546,
+    -2.8586881160736084,
+    -7.091681003570557,
+    -0.3582576513290405,
+    -2.6386361122131348,
+    -3.561924934387207,
+    1.672676682472229,
+    5.307150363922119,
+    2.5030627250671387,
+    0.7680068016052246,
+    -7.192819118499756,
+    -2.3492090702056885,
+    2.4020206928253174,
+    7.982364177703857,
+    -2.2059378623962402,
+    3.145641803741455,
+    5.874682426452637,
+    10.349384307861328,
+    -5.5676751136779785,
+    0.14949274063110352,
+    6.8305206298828125,
+    2.8054893016815186,
+    -3.023589611053467,
+    -2.727911949157715
+  ],
+  "sigma_frequencies": [
+    9.768982887268066,
+    -5.142672061920166,
+    -7.158572673797607,
+    3.8872931003570557,
+    0.783366322517395,
+    -3.6402578353881836,
+    12.029163360595703,
+    0.9776173830032349,
+    16.919240951538086,
+    2.1072747707366943,
+    2.3951961994171143,
+    5.7007927894592285,
+    3.5231542587280273,
+    -7.066169738769531,
+    -2.272156000137329,
+    -10.33111572265625,
+    -10.644245147705078,
+    1.4188770055770874,
+    -10.631746292114258,
+    0.99431312084198,
+    3.5392611026763916,
+    5.338685989379883,
+    0.41358622908592224,
+    -10.658740997314453,
+    8.450976371765137,
+    4.1102213859558105,
+    -1.0487929582595825,
+    -2.669220447540283,
+    -6.077164173126221,
+    1.115276575088501,
+    -1.5130574703216553,
+    -10.540763854980469,
+    9.292060852050781,
+    1.5696392059326172,
+    -9.652935028076172,
+    0.11881803721189499,
+    -4.144635200500488,
+    3.104952335357666,
+    -13.849717140197754,
+    0.5783001184463501,
+    2.3102877140045166,
+    9.099419593811035,
+    7.3769145011901855,
+    8.891700744628906,
+    -7.764804840087891,
+    -5.316822528839111,
+    -6.7786760330200195,
+    -8.151307106018066,
+    1.201075792312622,
+    2.4188919067382812,
+    -5.011605262756348,
+    -1.05170476436615,
+    -8.015609741210938,
+    -10.240509986877441,
+    -11.560667037963867,
+    -3.7953014373779297,
+    -1.0822583436965942,
+    4.821359634399414,
+    -0.08058752864599228,
+    -8.013398170471191,
+    3.7760136127471924,
+    -0.010018488392233849,
+    3.7555665969848633,
+    -4.78371000289917,
+    11.39970588684082,
+    10.100645065307617,
+    4.057767868041992,
+    2.469562530517578,
+    2.944430351257324,
+    -6.690946578979492,
+    -4.2867326736450195,
+    -4.587310314178467,
+    0.19896391034126282,
+    7.15863561630249,
+    1.8455699682235718,
+    -8.330734252929688,
+    -5.207420825958252,
+    -3.2296221256256104,
+    3.6857283115386963,
+    -9.234169006347656,
+    -2.4021103382110596,
+    12.515152931213379,
+    -1.5826791524887085,
+    2.2566795349121094,
+    0.8454548120498657,
+    14.971880912780762,
+    3.820692777633667,
+    4.668692588806152,
+    -9.071296691894531,
+    -7.492008209228516,
+    -0.2212979644536972,
+    -5.142780303955078,
+    -8.122804641723633,
+    0.11708371341228485,
+    -6.445141792297363,
+    11.834158897399902,
+    4.808426380157471,
+    -4.26067590713501,
+    -3.338383197784424,
+    -3.3120429515838623,
+    -12.90793514251709,
+    0.6678283214569092,
+    7.39938497543335,
+    -6.815516471862793,
+    -1.2518435716629028,
+    2.009228229522705,
+    -8.796148300170898,
+    -4.782722473144531,
+    -5.98541259765625,
+    1.3545910120010376,
+    -1.3095777034759521,
+    -9.345429420471191,
+    -2.6802685260772705,
+    -4.955610752105713,
+    -0.28176164627075195,
+    -9.338766098022461,
+    4.089934349060059,
+    3.968681812286377,
+    4.265690326690674,
+    7.729264259338379,
+    -5.696745872497559,
+    -1.1576746702194214,
+    -4.551684379577637,
+    1.423370599746704,
+    4.4033637046813965,
+    14.551665306091309,
+    -1.273985505104065,
+    -12.080366134643555,
+    3.688432455062866,
+    2.8932416439056396,
+    2.994927167892456,
+    -3.5851926803588867,
+    1.1186363697052002,
+    -0.023925336077809334,
+    2.5048391819000244,
+    -3.8339784145355225,
+    5.72321891784668,
+    2.3905227184295654,
+    -5.728427886962891,
+    -9.14192008972168,
+    -5.401311874389648,
+    0.09892720729112625,
+    -5.927880764007568,
+    2.1365606784820557,
+    -9.09748363494873,
+    -0.9202865362167358,
+    5.93142557144165,
+    -3.430886745452881,
+    2.375467538833618,
+    -2.820436477661133,
+    13.603337287902832,
+    -3.579545497894287,
+    0.4319279193878174,
+    -0.2505625784397125,
+    1.6767009496688843,
+    -5.083298683166504,
+    -3.666217088699341,
+    4.510189533233643,
+    14.891487121582031,
+    -2.894575834274292,
+    3.2277705669403076,
+    1.43055260181427,
+    -8.293705940246582,
+    8.418596267700195,
+    0.7782672047615051,
+    2.0863184928894043,
+    -4.327061176300049,
+    0.10986197739839554,
+    -7.969362258911133,
+    3.502122163772583,
+    4.26428747177124,
+    -5.891327381134033,
+    4.8794636726379395,
+    11.080082893371582,
+    -5.676795482635498,
+    11.000184059143066,
+    -3.6576757431030273,
+    -5.210165977478027,
+    3.798182249069214,
+    -7.83465051651001,
+    -6.119926452636719,
+    -6.116824626922607,
+    6.7045464515686035,
+    0.8135631680488586,
+    -0.5030490159988403,
+    7.793751239776611,
+    -0.6900088787078857,
+    -3.222320795059204,
+    -2.6430485248565674,
+    -1.8937172889709473,
+    -7.558099746704102,
+    5.619227409362793,
+    -1.745542287826538,
+    -8.842484474182129,
+    -5.361675262451172,
+    -2.745424270629883,
+    8.832099914550781,
+    -4.185833930969238,
+    8.491767883300781,
+    3.2348251342773438,
+    4.997406482696533,
+    -1.0009046792984009,
+    -6.707517147064209,
+    -5.479324817657471,
+    -9.24045181274414,
+    5.256299018859863,
+    0.7706965804100037,
+    8.959522247314453,
+    -8.60882568359375,
+    5.173210144042969,
+    6.907657146453857,
+    5.700893878936768,
+    0.5983768105506897,
+    -5.803370475769043,
+    8.23531436920166,
+    4.489255428314209,
+    3.3828928470611572,
+    -3.309412717819214,
+    -8.726313591003418,
+    -5.8629984855651855,
+    5.772271633148193,
+    -3.0476882457733154,
+    -0.16989658772945404,
+    -8.137606620788574,
+    0.4993816912174225,
+    -6.62803840637207,
+    8.565634727478027,
+    6.199972629547119,
+    9.28619384765625,
+    0.0005847634165547788,
+    -2.134239673614502,
+    -1.1670068502426147,
+    3.6982951164245605,
+    1.892624855041504,
+    -0.18664078414440155,
+    -1.3332287073135376,
+    -9.12190055847168,
+    -9.654301643371582,
+    -0.7440897822380066,
+    7.350038528442383,
+    9.781005859375,
+    -2.294945478439331,
+    -5.409823417663574,
+    17.33055877685547,
+    3.39593505859375,
+    8.776476860046387,
+    -14.440068244934082,
+    7.145877361297607,
+    -2.0074362754821777,
+    6.177412986755371,
+    -1.8212260007858276,
+    5.977794647216797,
+    0.4392381012439728,
+    1.3678476810455322,
+    6.645467758178711,
+    5.125301837921143,
+    -15.433785438537598,
+    1.72779381275177,
+    1.7806662321090698,
+    6.158847808837891,
+    -8.537556648254395,
+    0.009060980752110481,
+    -1.2375051975250244,
+    0.79191654920578,
+    7.771003246307373,
+    -1.8975887298583984,
+    0.914811909198761,
+    -1.7412867546081543,
+    -6.447122097015381,
+    7.294083118438721,
+    0.408783882856369,
+    -4.8634033203125,
+    -1.3467954397201538,
+    1.332945466041565,
+    0.7627815008163452,
+    2.337066888809204,
+    9.810226440429688,
+    -5.75670862197876,
+    8.774702072143555,
+    -7.363132476806641,
+    -13.072543144226074,
+    0.05342923849821091,
+    -13.84903335571289,
+    0.8227459788322449,
+    -7.357728481292725,
+    -2.4010021686553955,
+    7.029226779937744,
+    7.44602632522583,
+    10.075094223022461,
+    4.260078430175781,
+    5.6064958572387695,
+    -4.542119979858398,
+    1.132561206817627,
+    4.1948981285095215,
+    -2.711714029312134,
+    5.265280723571777,
+    -16.994571685791016,
+    11.43478775024414,
+    3.041484832763672,
+    2.2935965061187744,
+    -15.292680740356445,
+    -1.1755614280700684,
+    3.6270992755889893,
+    7.234384536743164,
+    -5.575100898742676,
+    -0.5625072121620178,
+    1.1739987134933472,
+    2.6426174640655518,
+    6.868399143218994,
+    -0.051543548703193665,
+    -2.4126954078674316,
+    -0.527182400226593,
+    1.3091306686401367,
+    -5.706780433654785,
+    -7.422503471374512,
+    16.04893684387207,
+    7.7499237060546875,
+    -8.368962287902832,
+    1.3969931602478027,
+    -0.1716773509979248,
+    -14.165465354919434,
+    -7.212979316711426,
+    -0.07946434617042542,
+    2.4266088008880615,
+    -1.6893504858016968,
+    -4.813210487365723,
+    7.460116386413574,
+    -2.8220160007476807,
+    -4.642193794250488,
+    -2.1517093181610107,
+    -10.031639099121094,
+    2.903050661087036,
+    14.23682975769043,
+    -4.212666988372803,
+    -8.54171085357666,
+    2.4117050170898438,
+    -11.86750602722168,
+    -2.2607569694519043,
+    10.836546897888184,
+    -3.936889886856079,
+    -4.861340522766113,
+    5.097802639007568,
+    -10.393189430236816,
+    -1.961990475654602,
+    3.8413000106811523,
+    -1.5600367784500122,
+    -0.5979607105255127,
+    0.15909065306186676,
+    6.979684829711914,
+    -0.9538497924804688,
+    -5.1925883293151855,
+    6.467310428619385,
+    -7.140716552734375,
+    -7.588068962097168,
+    -8.426490783691406,
+    8.518864631652832,
+    -2.828051805496216,
+    -9.814094543457031,
+    4.159399032592773,
+    13.71330738067627,
+    -3.5663294792175293,
+    -4.851027011871338,
+    0.7231943607330322,
+    8.881911277770996,
+    6.770598888397217,
+    -1.0106558799743652,
+    1.9355645179748535,
+    11.084402084350586,
+    0.3575453460216522,
+    -7.179681301116943,
+    -3.459652900695801,
+    -7.237196445465088,
+    -1.2357492446899414,
+    -5.271202564239502,
+    0.2657454311847687,
+    0.7321264147758484,
+    6.065614223480225,
+    7.138059139251709,
+    -5.723646640777588,
+    -0.10639117658138275,
+    -4.4325690269470215,
+    1.5119991302490234,
+    -1.5868138074874878,
+    3.202521324157715,
+    -1.394471287727356,
+    -1.0211089849472046,
+    -8.927684783935547,
+    -6.37411642074585,
+    -5.610091209411621,
+    -9.163214683532715,
+    8.649735450744629,
+    3.0920517444610596,
+    2.7999343872070312,
+    -7.620204925537109,
+    -4.139535427093506,
+    6.574669361114502,
+    -6.595216274261475,
+    5.857766628265381,
+    -0.43892958760261536,
+    -2.9875595569610596,
+    -7.38904333114624,
+    -8.716257095336914,
+    4.624642372131348,
+    -6.677160739898682,
+    -0.33922699093818665,
+    8.133686065673828,
+    3.1561405658721924,
+    2.721519708633423
+  ],
+  "tensors": {
+    "local.embedding": {
+      "offset": 0,
+      "bytes": 314976,
+      "shape": [
+        193,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.time_features": {
+      "offset": 314976,
+      "bytes": 1632,
+      "shape": [
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.time_embedding": {
+      "offset": 316608,
+      "bytes": 1331712,
+      "shape": [
+        408,
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.sigma_embedding": {
+      "offset": 1648320,
+      "bytes": 1331712,
+      "shape": [
+        408,
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.0.attention_norm.norm": {
+      "offset": 2980032,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.0.attention_norm.affine_weight": {
+      "offset": 2983296,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.0.attention_norm.affine_bias": {
+      "offset": 4315008,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.0.attention.projection": {
+      "offset": 4318272,
+      "bytes": 3290112,
+      "shape": [
+        2016,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.0.attention.qk_norm": {
+      "offset": 7608384,
+      "bytes": 272,
+      "shape": [
+        68
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.0.attention.head_weights": {
+      "offset": 7608656,
+      "bytes": 48,
+      "shape": [
+        12
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.0.attention.pair_bias": {
+      "offset": 7608704,
+      "bytes": 240,
+      "shape": [
+        12,
+        5
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.0.attention.output": {
+      "offset": 7608944,
+      "bytes": 1449216,
+      "shape": [
+        408,
+        888
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.0.ffn_norm.norm": {
+      "offset": 9058160,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.0.ffn_norm.affine_weight": {
+      "offset": 9061424,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.0.ffn_norm.affine_bias": {
+      "offset": 10393136,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.0.ffn.upgate": {
+      "offset": 10396400,
+      "bytes": 6658560,
+      "shape": [
+        4080,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.0.ffn.down": {
+      "offset": 17054960,
+      "bytes": 3329280,
+      "shape": [
+        408,
+        2040
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.1.attention_norm.norm": {
+      "offset": 20384240,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.1.attention_norm.affine_weight": {
+      "offset": 20387504,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.1.attention_norm.affine_bias": {
+      "offset": 21719216,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.1.attention.projection": {
+      "offset": 21722480,
+      "bytes": 3290112,
+      "shape": [
+        2016,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.1.attention.qk_norm": {
+      "offset": 25012592,
+      "bytes": 272,
+      "shape": [
+        68
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.1.attention.head_weights": {
+      "offset": 25012864,
+      "bytes": 48,
+      "shape": [
+        12
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.1.attention.pair_bias": {
+      "offset": 25012912,
+      "bytes": 240,
+      "shape": [
+        12,
+        5
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.1.attention.output": {
+      "offset": 25013152,
+      "bytes": 1449216,
+      "shape": [
+        408,
+        888
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.1.ffn_norm.norm": {
+      "offset": 26462368,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.1.ffn_norm.affine_weight": {
+      "offset": 26465632,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.1.ffn_norm.affine_bias": {
+      "offset": 27797344,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.1.ffn.upgate": {
+      "offset": 27800608,
+      "bytes": 6658560,
+      "shape": [
+        4080,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.1.ffn.down": {
+      "offset": 34459168,
+      "bytes": 3329280,
+      "shape": [
+        408,
+        2040
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.2.attention_norm.norm": {
+      "offset": 37788448,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.2.attention_norm.affine_weight": {
+      "offset": 37791712,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.2.attention_norm.affine_bias": {
+      "offset": 39123424,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.2.attention.projection": {
+      "offset": 39126688,
+      "bytes": 3290112,
+      "shape": [
+        2016,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.2.attention.qk_norm": {
+      "offset": 42416800,
+      "bytes": 272,
+      "shape": [
+        68
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.2.attention.head_weights": {
+      "offset": 42417072,
+      "bytes": 48,
+      "shape": [
+        12
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.2.attention.pair_bias": {
+      "offset": 42417120,
+      "bytes": 240,
+      "shape": [
+        12,
+        5
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.2.attention.output": {
+      "offset": 42417360,
+      "bytes": 1449216,
+      "shape": [
+        408,
+        888
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.2.ffn_norm.norm": {
+      "offset": 43866576,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.2.ffn_norm.affine_weight": {
+      "offset": 43869840,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.2.ffn_norm.affine_bias": {
+      "offset": 45201552,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.2.ffn.upgate": {
+      "offset": 45204816,
+      "bytes": 6658560,
+      "shape": [
+        4080,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.2.ffn.down": {
+      "offset": 51863376,
+      "bytes": 3329280,
+      "shape": [
+        408,
+        2040
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.3.attention_norm.norm": {
+      "offset": 55192656,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.3.attention_norm.affine_weight": {
+      "offset": 55195920,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.3.attention_norm.affine_bias": {
+      "offset": 56527632,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.3.attention.projection": {
+      "offset": 56530896,
+      "bytes": 3290112,
+      "shape": [
+        2016,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.3.attention.qk_norm": {
+      "offset": 59821008,
+      "bytes": 272,
+      "shape": [
+        68
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.3.attention.head_weights": {
+      "offset": 59821280,
+      "bytes": 48,
+      "shape": [
+        12
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.3.attention.pair_bias": {
+      "offset": 59821328,
+      "bytes": 240,
+      "shape": [
+        12,
+        5
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.3.attention.output": {
+      "offset": 59821568,
+      "bytes": 1449216,
+      "shape": [
+        408,
+        888
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.3.ffn_norm.norm": {
+      "offset": 61270784,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.3.ffn_norm.affine_weight": {
+      "offset": 61274048,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.3.ffn_norm.affine_bias": {
+      "offset": 62605760,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.3.ffn.upgate": {
+      "offset": 62609024,
+      "bytes": 6658560,
+      "shape": [
+        4080,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.3.ffn.down": {
+      "offset": 69267584,
+      "bytes": 3329280,
+      "shape": [
+        408,
+        2040
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.4.attention_norm.norm": {
+      "offset": 72596864,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.4.attention_norm.affine_weight": {
+      "offset": 72600128,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.4.attention_norm.affine_bias": {
+      "offset": 73931840,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.4.attention.projection": {
+      "offset": 73935104,
+      "bytes": 3290112,
+      "shape": [
+        2016,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.4.attention.qk_norm": {
+      "offset": 77225216,
+      "bytes": 272,
+      "shape": [
+        68
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.4.attention.head_weights": {
+      "offset": 77225488,
+      "bytes": 48,
+      "shape": [
+        12
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.4.attention.pair_bias": {
+      "offset": 77225536,
+      "bytes": 240,
+      "shape": [
+        12,
+        5
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.4.attention.output": {
+      "offset": 77225776,
+      "bytes": 1449216,
+      "shape": [
+        408,
+        888
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.4.ffn_norm.norm": {
+      "offset": 78674992,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.4.ffn_norm.affine_weight": {
+      "offset": 78678256,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.4.ffn_norm.affine_bias": {
+      "offset": 80009968,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.4.ffn.upgate": {
+      "offset": 80013232,
+      "bytes": 6658560,
+      "shape": [
+        4080,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.4.ffn.down": {
+      "offset": 86671792,
+      "bytes": 3329280,
+      "shape": [
+        408,
+        2040
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.5.attention_norm.norm": {
+      "offset": 90001072,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.5.attention_norm.affine_weight": {
+      "offset": 90004336,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.5.attention_norm.affine_bias": {
+      "offset": 91336048,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.5.attention.projection": {
+      "offset": 91339312,
+      "bytes": 3290112,
+      "shape": [
+        2016,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.5.attention.qk_norm": {
+      "offset": 94629424,
+      "bytes": 272,
+      "shape": [
+        68
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.5.attention.head_weights": {
+      "offset": 94629696,
+      "bytes": 48,
+      "shape": [
+        12
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.5.attention.pair_bias": {
+      "offset": 94629744,
+      "bytes": 240,
+      "shape": [
+        12,
+        5
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.5.attention.output": {
+      "offset": 94629984,
+      "bytes": 1449216,
+      "shape": [
+        408,
+        888
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.5.ffn_norm.norm": {
+      "offset": 96079200,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.5.ffn_norm.affine_weight": {
+      "offset": 96082464,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.5.ffn_norm.affine_bias": {
+      "offset": 97414176,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.5.ffn.upgate": {
+      "offset": 97417440,
+      "bytes": 6658560,
+      "shape": [
+        4080,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.5.ffn.down": {
+      "offset": 104076000,
+      "bytes": 3329280,
+      "shape": [
+        408,
+        2040
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.6.attention_norm.norm": {
+      "offset": 107405280,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.6.attention_norm.affine_weight": {
+      "offset": 107408544,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.6.attention_norm.affine_bias": {
+      "offset": 108740256,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.6.attention.projection": {
+      "offset": 108743520,
+      "bytes": 3290112,
+      "shape": [
+        2016,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.6.attention.qk_norm": {
+      "offset": 112033632,
+      "bytes": 272,
+      "shape": [
+        68
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.6.attention.head_weights": {
+      "offset": 112033904,
+      "bytes": 48,
+      "shape": [
+        12
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.6.attention.pair_bias": {
+      "offset": 112033952,
+      "bytes": 240,
+      "shape": [
+        12,
+        5
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.6.attention.output": {
+      "offset": 112034192,
+      "bytes": 1449216,
+      "shape": [
+        408,
+        888
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.6.ffn_norm.norm": {
+      "offset": 113483408,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.6.ffn_norm.affine_weight": {
+      "offset": 113486672,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.6.ffn_norm.affine_bias": {
+      "offset": 114818384,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.6.ffn.upgate": {
+      "offset": 114821648,
+      "bytes": 6658560,
+      "shape": [
+        4080,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.6.ffn.down": {
+      "offset": 121480208,
+      "bytes": 3329280,
+      "shape": [
+        408,
+        2040
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.7.attention_norm.norm": {
+      "offset": 124809488,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.7.attention_norm.affine_weight": {
+      "offset": 124812752,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.7.attention_norm.affine_bias": {
+      "offset": 126144464,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.7.attention.projection": {
+      "offset": 126147728,
+      "bytes": 3290112,
+      "shape": [
+        2016,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.7.attention.qk_norm": {
+      "offset": 129437840,
+      "bytes": 272,
+      "shape": [
+        68
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.7.attention.head_weights": {
+      "offset": 129438112,
+      "bytes": 48,
+      "shape": [
+        12
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.7.attention.pair_bias": {
+      "offset": 129438160,
+      "bytes": 240,
+      "shape": [
+        12,
+        5
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.7.attention.output": {
+      "offset": 129438400,
+      "bytes": 1449216,
+      "shape": [
+        408,
+        888
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.7.ffn_norm.norm": {
+      "offset": 130887616,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.7.ffn_norm.affine_weight": {
+      "offset": 130890880,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.7.ffn_norm.affine_bias": {
+      "offset": 132222592,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.7.ffn.upgate": {
+      "offset": 132225856,
+      "bytes": 6658560,
+      "shape": [
+        4080,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.7.ffn.down": {
+      "offset": 138884416,
+      "bytes": 3329280,
+      "shape": [
+        408,
+        2040
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.8.attention_norm.norm": {
+      "offset": 142213696,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.8.attention_norm.affine_weight": {
+      "offset": 142216960,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.8.attention_norm.affine_bias": {
+      "offset": 143548672,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.8.attention.projection": {
+      "offset": 143551936,
+      "bytes": 3290112,
+      "shape": [
+        2016,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.8.attention.qk_norm": {
+      "offset": 146842048,
+      "bytes": 272,
+      "shape": [
+        68
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.8.attention.head_weights": {
+      "offset": 146842320,
+      "bytes": 48,
+      "shape": [
+        12
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.8.attention.pair_bias": {
+      "offset": 146842368,
+      "bytes": 240,
+      "shape": [
+        12,
+        5
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.8.attention.output": {
+      "offset": 146842608,
+      "bytes": 1449216,
+      "shape": [
+        408,
+        888
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.8.ffn_norm.norm": {
+      "offset": 148291824,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.8.ffn_norm.affine_weight": {
+      "offset": 148295088,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.8.ffn_norm.affine_bias": {
+      "offset": 149626800,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.8.ffn.upgate": {
+      "offset": 149630064,
+      "bytes": 6658560,
+      "shape": [
+        4080,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.8.ffn.down": {
+      "offset": 156288624,
+      "bytes": 3329280,
+      "shape": [
+        408,
+        2040
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.9.attention_norm.norm": {
+      "offset": 159617904,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.9.attention_norm.affine_weight": {
+      "offset": 159621168,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.9.attention_norm.affine_bias": {
+      "offset": 160952880,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.9.attention.projection": {
+      "offset": 160956144,
+      "bytes": 3290112,
+      "shape": [
+        2016,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.9.attention.qk_norm": {
+      "offset": 164246256,
+      "bytes": 272,
+      "shape": [
+        68
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.9.attention.head_weights": {
+      "offset": 164246528,
+      "bytes": 48,
+      "shape": [
+        12
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.9.attention.pair_bias": {
+      "offset": 164246576,
+      "bytes": 240,
+      "shape": [
+        12,
+        5
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.9.attention.output": {
+      "offset": 164246816,
+      "bytes": 1449216,
+      "shape": [
+        408,
+        888
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.9.ffn_norm.norm": {
+      "offset": 165696032,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.9.ffn_norm.affine_weight": {
+      "offset": 165699296,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.9.ffn_norm.affine_bias": {
+      "offset": 167031008,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.9.ffn.upgate": {
+      "offset": 167034272,
+      "bytes": 6658560,
+      "shape": [
+        4080,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.9.ffn.down": {
+      "offset": 173692832,
+      "bytes": 3329280,
+      "shape": [
+        408,
+        2040
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.10.attention_norm.norm": {
+      "offset": 177022112,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.10.attention_norm.affine_weight": {
+      "offset": 177025376,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.10.attention_norm.affine_bias": {
+      "offset": 178357088,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.10.attention.projection": {
+      "offset": 178360352,
+      "bytes": 3290112,
+      "shape": [
+        2016,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.10.attention.qk_norm": {
+      "offset": 181650464,
+      "bytes": 272,
+      "shape": [
+        68
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.10.attention.head_weights": {
+      "offset": 181650736,
+      "bytes": 48,
+      "shape": [
+        12
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.10.attention.pair_bias": {
+      "offset": 181650784,
+      "bytes": 240,
+      "shape": [
+        12,
+        5
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.10.attention.output": {
+      "offset": 181651024,
+      "bytes": 1449216,
+      "shape": [
+        408,
+        888
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.10.ffn_norm.norm": {
+      "offset": 183100240,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.10.ffn_norm.affine_weight": {
+      "offset": 183103504,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.10.ffn_norm.affine_bias": {
+      "offset": 184435216,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.10.ffn.upgate": {
+      "offset": 184438480,
+      "bytes": 6658560,
+      "shape": [
+        4080,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.10.ffn.down": {
+      "offset": 191097040,
+      "bytes": 3329280,
+      "shape": [
+        408,
+        2040
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.11.attention_norm.norm": {
+      "offset": 194426320,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.11.attention_norm.affine_weight": {
+      "offset": 194429584,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.11.attention_norm.affine_bias": {
+      "offset": 195761296,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.11.attention.projection": {
+      "offset": 195764560,
+      "bytes": 3290112,
+      "shape": [
+        2016,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.11.attention.qk_norm": {
+      "offset": 199054672,
+      "bytes": 272,
+      "shape": [
+        68
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.11.attention.head_weights": {
+      "offset": 199054944,
+      "bytes": 48,
+      "shape": [
+        12
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.11.attention.pair_bias": {
+      "offset": 199054992,
+      "bytes": 240,
+      "shape": [
+        12,
+        5
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.11.attention.output": {
+      "offset": 199055232,
+      "bytes": 1449216,
+      "shape": [
+        408,
+        888
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.11.ffn_norm.norm": {
+      "offset": 200504448,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.11.ffn_norm.affine_weight": {
+      "offset": 200507712,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.11.ffn_norm.affine_bias": {
+      "offset": 201839424,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.11.ffn.upgate": {
+      "offset": 201842688,
+      "bytes": 6658560,
+      "shape": [
+        4080,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.blocks.11.ffn.down": {
+      "offset": 208501248,
+      "bytes": 3329280,
+      "shape": [
+        408,
+        2040
+      ],
+      "dtype": "float32"
+    },
+    "local.endpoint_updates.0": {
+      "offset": 211830528,
+      "bytes": 4896,
+      "shape": [
+        3,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.endpoint_updates.1": {
+      "offset": 211835424,
+      "bytes": 4896,
+      "shape": [
+        3,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "local.endpoint_updates.2": {
+      "offset": 211840320,
+      "bytes": 4896,
+      "shape": [
+        3,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite_time_embedding": {
+      "offset": 211845216,
+      "bytes": 3995136,
+      "shape": [
+        408,
+        2448
+      ],
+      "dtype": "float32"
+    },
+    "noise_encoder": {
+      "offset": 215840352,
+      "bytes": 9792,
+      "shape": [
+        408,
+        6
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.0.attention_norm.norm": {
+      "offset": 215850144,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.0.attention_norm.affine_weight": {
+      "offset": 215853408,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.0.attention_norm.affine_bias": {
+      "offset": 217185120,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.0.attention.projection": {
+      "offset": 217188384,
+      "bytes": 3290112,
+      "shape": [
+        2016,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.0.attention.qk_norm": {
+      "offset": 220478496,
+      "bytes": 272,
+      "shape": [
+        68
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.0.attention.head_weights": {
+      "offset": 220478768,
+      "bytes": 48,
+      "shape": [
+        12
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.0.attention.pair_bias": {
+      "offset": 220478816,
+      "bytes": 240,
+      "shape": [
+        12,
+        5
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.0.attention.output": {
+      "offset": 220479056,
+      "bytes": 1449216,
+      "shape": [
+        408,
+        888
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.0.ffn_norm.norm": {
+      "offset": 221928272,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.0.ffn_norm.affine_weight": {
+      "offset": 221931536,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.0.ffn_norm.affine_bias": {
+      "offset": 223263248,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.0.ffn.upgate": {
+      "offset": 223266512,
+      "bytes": 6658560,
+      "shape": [
+        4080,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.0.ffn.down": {
+      "offset": 229925072,
+      "bytes": 3329280,
+      "shape": [
+        408,
+        2040
+      ],
+      "dtype": "float32"
+    },
+    "finite.lateral.outputs.0": {
+      "offset": 233254352,
+      "bytes": 1671168,
+      "shape": [
+        408,
+        1024
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.1.attention_norm.norm": {
+      "offset": 234925520,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.1.attention_norm.affine_weight": {
+      "offset": 234928784,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.1.attention_norm.affine_bias": {
+      "offset": 236260496,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.1.attention.projection": {
+      "offset": 236263760,
+      "bytes": 3290112,
+      "shape": [
+        2016,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.1.attention.qk_norm": {
+      "offset": 239553872,
+      "bytes": 272,
+      "shape": [
+        68
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.1.attention.head_weights": {
+      "offset": 239554144,
+      "bytes": 48,
+      "shape": [
+        12
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.1.attention.pair_bias": {
+      "offset": 239554192,
+      "bytes": 240,
+      "shape": [
+        12,
+        5
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.1.attention.output": {
+      "offset": 239554432,
+      "bytes": 1449216,
+      "shape": [
+        408,
+        888
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.1.ffn_norm.norm": {
+      "offset": 241003648,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.1.ffn_norm.affine_weight": {
+      "offset": 241006912,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.1.ffn_norm.affine_bias": {
+      "offset": 242338624,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.1.ffn.upgate": {
+      "offset": 242341888,
+      "bytes": 6658560,
+      "shape": [
+        4080,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.1.ffn.down": {
+      "offset": 249000448,
+      "bytes": 3329280,
+      "shape": [
+        408,
+        2040
+      ],
+      "dtype": "float32"
+    },
+    "finite.lateral.outputs.1": {
+      "offset": 252329728,
+      "bytes": 1671168,
+      "shape": [
+        408,
+        1024
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.2.attention_norm.norm": {
+      "offset": 254000896,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.2.attention_norm.affine_weight": {
+      "offset": 254004160,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.2.attention_norm.affine_bias": {
+      "offset": 255335872,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.2.attention.projection": {
+      "offset": 255339136,
+      "bytes": 3290112,
+      "shape": [
+        2016,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.2.attention.qk_norm": {
+      "offset": 258629248,
+      "bytes": 272,
+      "shape": [
+        68
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.2.attention.head_weights": {
+      "offset": 258629520,
+      "bytes": 48,
+      "shape": [
+        12
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.2.attention.pair_bias": {
+      "offset": 258629568,
+      "bytes": 240,
+      "shape": [
+        12,
+        5
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.2.attention.output": {
+      "offset": 258629808,
+      "bytes": 1449216,
+      "shape": [
+        408,
+        888
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.2.ffn_norm.norm": {
+      "offset": 260079024,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.2.ffn_norm.affine_weight": {
+      "offset": 260082288,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.2.ffn_norm.affine_bias": {
+      "offset": 261414000,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.2.ffn.upgate": {
+      "offset": 261417264,
+      "bytes": 6658560,
+      "shape": [
+        4080,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.2.ffn.down": {
+      "offset": 268075824,
+      "bytes": 3329280,
+      "shape": [
+        408,
+        2040
+      ],
+      "dtype": "float32"
+    },
+    "finite.lateral.outputs.2": {
+      "offset": 271405104,
+      "bytes": 1671168,
+      "shape": [
+        408,
+        1024
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.3.attention_norm.norm": {
+      "offset": 273076272,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.3.attention_norm.affine_weight": {
+      "offset": 273079536,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.3.attention_norm.affine_bias": {
+      "offset": 274411248,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.3.attention.projection": {
+      "offset": 274414512,
+      "bytes": 3290112,
+      "shape": [
+        2016,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.3.attention.qk_norm": {
+      "offset": 277704624,
+      "bytes": 272,
+      "shape": [
+        68
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.3.attention.head_weights": {
+      "offset": 277704896,
+      "bytes": 48,
+      "shape": [
+        12
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.3.attention.pair_bias": {
+      "offset": 277704944,
+      "bytes": 240,
+      "shape": [
+        12,
+        5
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.3.attention.output": {
+      "offset": 277705184,
+      "bytes": 1449216,
+      "shape": [
+        408,
+        888
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.3.ffn_norm.norm": {
+      "offset": 279154400,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.3.ffn_norm.affine_weight": {
+      "offset": 279157664,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.3.ffn_norm.affine_bias": {
+      "offset": 280489376,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.3.ffn.upgate": {
+      "offset": 280492640,
+      "bytes": 6658560,
+      "shape": [
+        4080,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.3.ffn.down": {
+      "offset": 287151200,
+      "bytes": 3329280,
+      "shape": [
+        408,
+        2040
+      ],
+      "dtype": "float32"
+    },
+    "finite.lateral.outputs.3": {
+      "offset": 290480480,
+      "bytes": 1671168,
+      "shape": [
+        408,
+        1024
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.4.attention_norm.norm": {
+      "offset": 292151648,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.4.attention_norm.affine_weight": {
+      "offset": 292154912,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.4.attention_norm.affine_bias": {
+      "offset": 293486624,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.4.attention.projection": {
+      "offset": 293489888,
+      "bytes": 3290112,
+      "shape": [
+        2016,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.4.attention.qk_norm": {
+      "offset": 296780000,
+      "bytes": 272,
+      "shape": [
+        68
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.4.attention.head_weights": {
+      "offset": 296780272,
+      "bytes": 48,
+      "shape": [
+        12
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.4.attention.pair_bias": {
+      "offset": 296780320,
+      "bytes": 240,
+      "shape": [
+        12,
+        5
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.4.attention.output": {
+      "offset": 296780560,
+      "bytes": 1449216,
+      "shape": [
+        408,
+        888
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.4.ffn_norm.norm": {
+      "offset": 298229776,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.4.ffn_norm.affine_weight": {
+      "offset": 298233040,
+      "bytes": 1331712,
+      "shape": [
+        816,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.4.ffn_norm.affine_bias": {
+      "offset": 299564752,
+      "bytes": 3264,
+      "shape": [
+        816
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.4.ffn.upgate": {
+      "offset": 299568016,
+      "bytes": 6658560,
+      "shape": [
+        4080,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.blocks.4.ffn.down": {
+      "offset": 306226576,
+      "bytes": 3329280,
+      "shape": [
+        408,
+        2040
+      ],
+      "dtype": "float32"
+    },
+    "finite.lateral.outputs.4": {
+      "offset": 309555856,
+      "bytes": 1671168,
+      "shape": [
+        408,
+        1024
+      ],
+      "dtype": "float32"
+    },
+    "finite.lateral.projections.0": {
+      "offset": 311227024,
+      "bytes": 1671168,
+      "shape": [
+        1024,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.lateral.projections.1": {
+      "offset": 312898192,
+      "bytes": 1671168,
+      "shape": [
+        1024,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.lateral.projections.2": {
+      "offset": 314569360,
+      "bytes": 1671168,
+      "shape": [
+        1024,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.lateral.projections.3": {
+      "offset": 316240528,
+      "bytes": 1671168,
+      "shape": [
+        1024,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.lateral.projections.4": {
+      "offset": 317911696,
+      "bytes": 1671168,
+      "shape": [
+        1024,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.lateral.projections.5": {
+      "offset": 319582864,
+      "bytes": 1671168,
+      "shape": [
+        1024,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.lateral.projections.6": {
+      "offset": 321254032,
+      "bytes": 1671168,
+      "shape": [
+        1024,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.lateral.projections.7": {
+      "offset": 322925200,
+      "bytes": 1671168,
+      "shape": [
+        1024,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.lateral.projections.8": {
+      "offset": 324596368,
+      "bytes": 1671168,
+      "shape": [
+        1024,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.lateral.projections.9": {
+      "offset": 326267536,
+      "bytes": 1671168,
+      "shape": [
+        1024,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.lateral.projections.10": {
+      "offset": 327938704,
+      "bytes": 1671168,
+      "shape": [
+        1024,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.lateral.projections.11": {
+      "offset": 329609872,
+      "bytes": 1671168,
+      "shape": [
+        1024,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.lateral.mixing": {
+      "offset": 331281040,
+      "bytes": 245760,
+      "shape": [
+        5,
+        12,
+        1024
+      ],
+      "dtype": "float32"
+    },
+    "finite.endpoint_updates.0": {
+      "offset": 331526800,
+      "bytes": 4896,
+      "shape": [
+        3,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.endpoint_updates.1": {
+      "offset": 331531696,
+      "bytes": 4896,
+      "shape": [
+        3,
+        408
+      ],
+      "dtype": "float32"
+    },
+    "finite.endpoint_updates.2": {
+      "offset": 331536592,
+      "bytes": 4896,
+      "shape": [
+        3,
+        408
+      ],
+      "dtype": "float32"
+    }
+  }
+}

webgpu/v8_ck_240000/weights.f32

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