Repository: huggingface/transformers
Pull request: Add XPU Expectations for vibe voice acoustic tokenizer tests

@IlyasMoutawwakil pls help review, thx!

Changed files:
filename: tests/fixtures/vibevoice/expected_acoustic_tokenizer_results.json
patch: @@ -1 +1,4 @@
-{"encoder": [[2.140625, 3.0625, -1.3671875, 5.09375, -6.125, -11.125, 0.007080078125, 0.62890625, -0.263671875, 2.734375, 3.171875, 4.8125, 1.421875, -5.125, -0.431640625, 2.3125, -5.71875, 1.296875, 5.375, 4.40625, -1.1640625, 15.5625, -3.640625, 2.390625, 1.796875, 3.484375, 0.46484375, 1.5, 0.3359375, -4.28125, 2.875, -0.4375, 2.71875, 2.859375, -2.125, -5.09375, -2.375, -4.28125, 3.96875, 10.375, 1.09375, 7.96875, 12.5, -3.46875, 0.5234375, 0.380859375, -2.046875, -7.1875, 1.5625, -6.46875, 3.1875, 0.236328125, -1.0390625, 4.21875, 2.234375, -11.5, -1.9921875, -4.34375, 1.4609375, -5.96875, -2.203125, 2.515625, 5.71875, 0.357421875, -0.62109375, -3.15625, -15.1875, -2.546875, 7.53125, -2.46875, -0.30078125, 2.828125, 8.4375, 0.6796875, 3.53125, 0.48828125, 2.203125, 0.671875, 4.9375, -3.34375, 2.578125, 4.375, 1.7578125, 11.9375, -4.6875, 5.125, -1.90625, -9.875, 4.21875, 0.92578125, -9.375, 1.7890625, 2.296875, -6.65625, 1.8828125, -1.625, 3.265625, 12.1875, -1.4296875, -0.5234375, -4.90625, -1.328125, -1.2890625, -9.25, -0.384765625, -5.71875, 0.197265625, 6.375, 5.6875, 3.046875, 6.21875, 8.625, -3.0625, -7.40625, 5.0625, -4.5, -13.3125, -1.65625, 3.421875, 2.875, -0.5546875, 4.34375, 2.453125, -3.609375, -8.1875, -1.28125, 4.84375, 1.21875, 10.75, -0.5625, -5.3125, -3.375, 3.25, 16.125, 10.9375, -8.25, 8.625, 5.4375, -0.51171875, -8.375, 10.0625, 6.71875, 1.9140625, -0.65625, -10.9375, -0.07763671875, 5.75, -3.71875, 6.6875, -0.9296875, 14.0625, -6.0, -8.4375, -6.875, 7.9375, -4.75, -3.921875, -6.46875, 8.0625, -0.11328125, -3.296875, 8.25, 7.71875, 0.48828125, -10.1875, -2.953125, 2.75, -4.40625, 6.34375, 14.8125, -9.5, -18.25, 11.3125, 4.96875, -2.34375, 0.7265625, -1.1953125, 7.5, -2.859375, -4.5, -2.296875, 5.03125, -0.83984375, -6.75, 7.40625, -4.375, 0.39453125, 6.1875, -5.71875, 5.8125, -4.75, 11.9375, -7.5625, 3.75, -2.140625, 3.984375, 5.40625, 7.09375, 4.5625, 3.109375], [1.4609375, 1.0234375, -2.828125, 3.5, -4.78125, -2.5625, 1.21875, 1.046875, 4.875, 1.03125, -0.06640625, -2.9375, 2.125, 0.9921875, -1.3046875, -0.1787109375, 5.28125, -1.3828125, -0.3828125, -2.1875, 0.390625, 0.4140625, 3.359375, 8.125, -1.0078125, -4.59375, -2.640625, -0.1611328125, -0.8125, -3.328125, 1.0859375, 2.359375, -0.404296875, -0.5859375, -2.09375, 0.359375, 2.5625, 3.453125, 2.53125, 3.328125, -0.6015625, 1.9140625, 0.9453125, -2.96875, -3.953125, -0.4296875, 4.59375, 2.5625, -0.326171875, 2.359375, -0.458984375, 2.546875, -5.84375, 2.046875, -4.4375, -1.5390625, 3.046875, 1.734375, -1.5703125, -2.0625, -1.8359375, 0.75, -0.84765625, 0.486328125, -0.625, 0.7578125, -0.5859375, -2.640625, -1.2109375, 0.609375, 3.625, -4.59375, 2.75, -2.703125, 1.5, -0.57421875, 0.19140625, -1.6640625, -1.453125, 2.828125, 3.265625, 2.375, -1.8515625, -0.515625, -0.515625, 2.03125, 1.5, 0.90234375, -0.60546875, -0.72265625, 2.703125, -2.046875, -3.8125, -0.08056640625, -1.40625, -2.21875, -2.609375, -2.0625, -1.3046875, -0.134765625, -1.703125, -2.3125, 1.6171875, 0.1982421875, 0.95703125, 1.2265625, -1.125, -0.7734375, -0.375, -0.9609375, -4.21875, 0.42578125, 1.53125, -2.015625, 1.6875, 1.1171875, -0.6171875, 2.390625, 1.25, 0.3359375, 2.796875, 1.3203125, -1.5546875, -2.203125, 0.46484375, -0.35546875, -3.234375, -0.98828125, 0.6484375, -0.5546875, 0.4921875, -2.296875, -2.515625, 0.8671875, -0.0771484375, -0.69140625, 0.9453125, 0.52734375, 0.91796875, -1.5078125, -2.421875, -1.7890625, -1.5, 1.9375, 0.8671875, -2.265625, -1.46875, -2.765625, 2.453125, 1.078125, 3.0, 1.59375, 0.328125, -3.0625, -0.5546875, -1.296875, -1.609375, 0.3203125, -1.515625, -1.0703125, -5.65625, -0.734375, 2.34375, 0.81640625, -1.7890625, -2.125, 0.78125, 2.609375, 2.1875, 4.84375, 1.671875, 1.6484375, 0.462890625, -0.3984375, -1.8984375, -3.6875, 0.030517578125, -2.21875, -1.328125, -3.015625, 2.734375, 3.609375, 1.0234375, -1.1640625, -2.328125, -1.0625, -1.7578125, -0.6640625, 0.98046875, -1.2578125, 0.4609375, -1.9375, -0.046875, -3.25, -2.96875, -3.78125, -3.359375, -0.51171875, 2.828125, -2.828125]], "decoder": [[[-2.205371856689453e-06, 2.384185791015625e-06, 6.318092346191406e-06, 4.827976226806641e-06, -7.927417755126953e-06, -9.47713851928711e-06, 7.152557373046875e-06, 8.165836334228516e-06, -3.3080577850341797e-06, -1.3113021850585938e-06, 5.125999450683594e-06, -5.781650543212891e-06, -3.129243850708008e-06, 7.510185241699219e-06, 1.7881393432617188e-06, 3.874301910400391e-06, -5.930662155151367e-06, -5.036592483520508e-06, -1.5795230865478516e-06, 6.556510925292969e-07, 2.8014183044433594e-06, -4.470348358154297e-07, -1.6093254089355469e-06, -3.2782554626464844e-07, 2.4139881134033203e-06, 9.5367431640625e-07, 8.046627044677734e-07, -1.2516975402832031e-06, -8.940696716308594e-08, -2.5480985641479492e-06, -2.294778823852539e-06, -8.195638656616211e-07, -9.5367431640625e-07, 1.0132789611816406e-06, 1.1324882507324219e-06, 8.940696716308594e-08, -1.087784767150879e-06, -2.60770320892334e-06, -1.3113021850585938e-06, -9.238719940185547e-07, 2.980232238769531e-07, 0.0, 1.3113021850585938e-06, 6.258487701416016e-07, 9.238719940185547e-07, -4.172325134277344e-07, -3.2782554626464844e-07, 1.341104507446289e-06, 5.364418029785156e-07, -5.960464477539062e-07, -5.960464477539063e-08, -2.682209014892578e-07, 2.9802322387695312e-08, -7.301568984985352e-07, 6.556510925292969e-07, 6.854534149169922e-07, -2.384185791015625e-07, -6.854534149169922e-07, -8.046627044677734e-07, 4.470348358154297e-07, 4.172325134277344e-07, 6.556510925292969e-07, 0.0, 4.470348358154297e-07, 3.2782554626464844e-07, -8.940696716308594e-08, 8.940696716308594e-08, 5.364418029785156e-07, 6.854534149169922e-07, -1.4901161193847656e-07, -2.9802322387695312e-08, 2.980232238769531e-07, 2.682209014892578e-07, -1.4901161193847656e-07, 2.9802322387695312e-08, -3.8743019104003906e-07, -6.854534149169922e-07, -5.066394805908203e-07, -2.384185791015625e-07, 1.1920928955078125e-07, 1.1920928955078125e-07, -2.980232238769531e-07, -1.7881393432617188e-07, 3.2782554626464844e-07, 5.364418029785156e-07, 1.7881393432617188e-07, 0.0, 3.8743019104003906e-07, 0.0, -2.980232238769531e-07, 0.0, 7.450580596923828e-07, 6.258487701416016e-07, 2.9802322387695312e-08, 3.2782554626464844e-07, 5.364418029785156e-07, 3.2782554626464844e-07, -5.960464477539063e-08, -5.960464477539063e-08, 4.172325134277344e-07, 1.4901161193847656e-07, -4.172325134277344e-07, -2.9802322387695312e-08, 4.470348358154297e-07, 3.2782554626464844e-07, -5.066394805908203e-07, -8.940696716308594e-08, 4.76837158203125e-07, -5.960464477539063e-08, -4.76837158203125e-07, -3.5762786865234375e-07, 2.682209014892578e-07, 1.1920928955078125e-07, -6.258487701416016e-07, -2.384185791015625e-07, 1.1920928955078125e-06, 5.662441253662109e-07, -2.086162567138672e-07, -2.086162567138672e-07, 1.4901161193847656e-07, -8.940696716308594e-08, -8.046627044677734e-07, -4.76837158203125e-07, -2.086162567138672e-07, -8.940696716308594e-08, -8.940696716308594e-08, 1.7881393432617188e-07, 7.152557373046875e-07, -1.1920928955078125e-07, -7.450580596923828e-07, -2.086162567138672e-07, 3.5762786865234375e-07, 1.4901161193847656e-07, -1.4901161193847656e-07, 2.9802322387695312e-08, 2.682209014892578e-07, 0.0, -2.980232238769531e-07, -8.940696716308594e-08, -1.1920928955078125e-07, -2.384185791015625e-07, -8.940696716308594e-08, 2.384185791015625e-07, 4.470348358154297e-07, 2.384185791015625e-07, -3.5762786865234375e-07, -3.5762786865234375e-07, 2.980232238769531e-07, 2.384185791015625e-07, 4.76837158203125e-07, 3.8743019104003906e-07, 2.384185791015625e-07, 2.9802322387695312e-08, -1.7881393432617188e-07, 1.4901161193847656e-07, 3.8743019104003906e-07, 1.1920928955078125e-07, -8.940696716308594e-08, 0.0, 2.682209014892578e-07, 0.0, -2.980232238769531e-07, -1.1920928955078125e-07, 4.172325134277344e-07, 5.364418029785156e-07, 1.4901161193847656e-07, 3.8743019104003906e-07, 2.682209014892578e-07, 0.0, -3.8743019104003906e-07, -1.1920928955078125e-07, 2.980232238769531e-07, 2.384185791015625e-07, 2.384185791015625e-07, 2.9802322387695312e-08, 4.470348358154297e-07, 0.0, 2.9802322387695312e-08, 2.682209014892578e-07, 8.940696716308594e-08, 2.086162567138672e-07, -2.384185791015625e-07, -4.172325134277344e-07, 2.384185791015625e-07, 2.9802322387695312e-08, -4.76837158203125e-07, 0.0, 5.066394805908203e-07, 3.5762786865234375e-07, -1.7881393432617188e-07, -1.7881393432617188e-07, 3.5762786865234375e-07, 6.854534149169922e-07, 1.4901161193847656e-07, 1.7881393432617188e-07, 7.748603820800781e-07, 1.7881393432617188e-07, -5.066394805908203e-07, -1.1920928955078125e-07, 3.8743019104003906e-07]], [[-2.5033950805664062e-05, -3.743171691894531e-05, -8.20159912109375e-05, -0.00018024444580078125, -0.00029754638671875, -0.000274658203125, -0.00013637542724609375, -9.298324584960938e-06, 8.678436279296875e-05, 0.00021076202392578125, 0.00022411346435546875, 0.00012111663818359375, 3.933906555175781e-05, 2.7418136596679688e-05, 6.4849853515625e-05, 0.000171661376953125, 0.000125885009765625, -5.21540641784668e-06, -0.00018596649169921875, -0.0002040863037109375, -9.632110595703125e-05, -3.647804260253906e-05, -1.4603137969970703e-05, -2.944469451904297e-05, -1.424551010131836e-05, 8.678436279296875e-05, 0.0002956390380859375, 0.000423431396484375, 0.000331878662109375, 0.000255584716796875, 0.00011730194091796875, -0.0001201629638671875, -0.00021076202392578125, -0.0001926422119140625, -0.000301361083984375, -0.000492095947265625, -0.0005340576171875, -0.0003566741943359375, -0.000179290771484375, -8.487701416015625e-05, -3.9577484130859375e-05, -1.5854835510253906e-05, -0.00010776519775390625, -0.0002460479736328125, -0.0002803802490234375, -0.000324249267578125, -0.0002288818359375, -8.58306884765625e-05, -0.00012683868408203125, -0.0001430511474609375, -3.62396240234375e-05, -3.4332275390625e-05, -0.000156402587890625, -0.000240325927734375, -0.00031280517578125, -0.000335693359375, -0.0003986358642578125, -0.0004711151123046875, -0.0003108978271484375, -7.915496826171875e-05, 0.00010776519775390625, 3.600120544433594e-05, -0.00014781951904296875, -0.0004138946533203125, -0.000583648681640625, -0.00048065185546875, -0.000308990478515625, -0.000316619873046875, -0.0003147125244140625, -0.0004024505615234375, -0.000629425048828125, -0.000591278076171875, -0.0002899169921875, 0.00012969970703125, 0.0004177093505859375, 0.0003986358642578125, 0.0003299713134765625, 0.0002918243408203125, 0.000202178955078125, 0.0001010894775390625, -3.5762786865234375e-05, -0.000179290771484375, -0.00023937225341796875, -6.580352783203125e-05, 7.82012939453125e-05, -0.00011539459228515625, -0.00043487548828125, -0.000591278076171875, -0.0002574920654296875, 0.0003185272216796875, 0.000537872314453125, 0.000457763671875, 0.000335693359375, 9.250640869140625e-05, -0.0001735687255859375, -0.0002536773681640625, -0.0003643035888671875, -0.00048065185546875, -0.0003795623779296875, -0.000370025634765625, -0.0004596710205078125, -0.000507354736328125, -0.00034332275390625, -5.340576171875e-05, 0.000274658203125, 0.000438690185546875, 0.00034332275390625, 9.34600830078125e-05, -3.4570693969726562e-06, -2.771615982055664e-06, -3.0994415283203125e-05, -6.031990051269531e-05, -0.00018596649169921875, -0.0003509521484375, -0.0003032684326171875, -0.0001277923583984375, 5.4836273193359375e-05, 0.0002651214599609375, 0.000492095947265625, 0.000591278076171875, 0.00058746337890625, 0.000614166259765625, 0.00061798095703125, 0.00049591064453125, 0.0002574920654296875, -9.98377799987793e-07, -0.00019550323486328125, -0.000263214111328125, -0.00013065338134765625, 0.00017452239990234375, 0.0003719329833984375, 0.0004119873046875, 0.0002841949462890625, 0.00022602081298828125, 2.9325485229492188e-05, -3.933906555175781e-05, 0.000110626220703125, 0.00014972686767578125, 4.00543212890625e-05, -0.00015735626220703125, -0.0003452301025390625, -0.0003871917724609375, -7.724761962890625e-05, 0.00040435791015625, 0.000469207763671875, 0.0001983642578125, 5.316734313964844e-05, -0.00012493133544921875, -0.00030517578125, -0.0004730224609375, -0.0004482269287109375, -0.0004520416259765625, -0.0006256103515625, -0.000720977783203125, -0.000812530517578125, -0.000713348388671875, -0.0004673004150390625, -0.00011730194091796875, 0.00016880035400390625, 0.000324249267578125, 0.0002498626708984375, 0.0002498626708984375, 0.000331878662109375, 0.0001544952392578125, -0.000270843505859375, -0.000690460205078125, -0.000713348388671875, -0.000461578369140625, -0.0002956390380859375, -0.00018978118896484375, -3.647804260253906e-05, 0.000118255615234375, 0.000232696533203125, 0.0002918243408203125, 0.0002803802490234375, 0.00017833709716796875, 1.1920928955078125e-05, -0.000240325927734375, -0.0002994537353515625, -0.00020885467529296875, -0.0002899169921875, -0.000446319580078125, -0.0004444122314453125, -0.0005035400390625, -0.000484466552734375, -0.0002574920654296875, -2.9802322387695312e-05, 0.00017070770263671875, 0.00016117095947265625, 4.291534423828125e-05, -0.0001430511474609375, -0.0003147125244140625, -0.00019741058349609375, 9.1552734375e-05, 0.0002002716064453125, 0.0001163482666015625, 5.888938903808594e-05, -9.822845458984375e-05, -0.0001354217529296875, -4.1961669921875e-05]]]}
\ No newline at end of file
+{
+    "cuda": {"encoder": [[2.140625, 3.0625, -1.3671875, 5.09375, -6.125, -11.125, 0.007080078125, 0.62890625, -0.263671875, 2.734375, 3.171875, 4.8125, 1.421875, -5.125, -0.431640625, 2.3125, -5.71875, 1.296875, 5.375, 4.40625, -1.1640625, 15.5625, -3.640625, 2.390625, 1.796875, 3.484375, 0.46484375, 1.5, 0.3359375, -4.28125, 2.875, -0.4375, 2.71875, 2.859375, -2.125, -5.09375, -2.375, -4.28125, 3.96875, 10.375, 1.09375, 7.96875, 12.5, -3.46875, 0.5234375, 0.380859375, -2.046875, -7.1875, 1.5625, -6.46875, 3.1875, 0.236328125, -1.0390625, 4.21875, 2.234375, -11.5, -1.9921875, -4.34375, 1.4609375, -5.96875, -2.203125, 2.515625, 5.71875, 0.357421875, -0.62109375, -3.15625, -15.1875, -2.546875, 7.53125, -2.46875, -0.30078125, 2.828125, 8.4375, 0.6796875, 3.53125, 0.48828125, 2.203125, 0.671875, 4.9375, -3.34375, 2.578125, 4.375, 1.7578125, 11.9375, -4.6875, 5.125, -1.90625, -9.875, 4.21875, 0.92578125, -9.375, 1.7890625, 2.296875, -6.65625, 1.8828125, -1.625, 3.265625, 12.1875, -1.4296875, -0.5234375, -4.90625, -1.328125, -1.2890625, -9.25, -0.384765625, -5.71875, 0.197265625, 6.375, 5.6875, 3.046875, 6.21875, 8.625, -3.0625, -7.40625, 5.0625, -4.5, -13.3125, -1.65625, 3.421875, 2.875, -0.5546875, 4.34375, 2.453125, -3.609375, -8.1875, -1.28125, 4.84375, 1.21875, 10.75, -0.5625, -5.3125, -3.375, 3.25, 16.125, 10.9375, -8.25, 8.625, 5.4375, -0.51171875, -8.375, 10.0625, 6.71875, 1.9140625, -0.65625, -10.9375, -0.07763671875, 5.75, -3.71875, 6.6875, -0.9296875, 14.0625, -6.0, -8.4375, -6.875, 7.9375, -4.75, -3.921875, -6.46875, 8.0625, -0.11328125, -3.296875, 8.25, 7.71875, 0.48828125, -10.1875, -2.953125, 2.75, -4.40625, 6.34375, 14.8125, -9.5, -18.25, 11.3125, 4.96875, -2.34375, 0.7265625, -1.1953125, 7.5, -2.859375, -4.5, -2.296875, 5.03125, -0.83984375, -6.75, 7.40625, -4.375, 0.39453125, 6.1875, -5.71875, 5.8125, -4.75, 11.9375, -7.5625, 3.75, -2.140625, 3.984375, 5.40625, 7.09375, 4.5625, 3.109375], [1.4609375, 1.0234375, -2.828125, 3.5, -4.78125, -2.5625, 1.21875, 1.046875, 4.875, 1.03125, -0.06640625, -2.9375, 2.125, 0.9921875, -1.3046875, -0.1787109375, 5.28125, -1.3828125, -0.3828125, -2.1875, 0.390625, 0.4140625, 3.359375, 8.125, -1.0078125, -4.59375, -2.640625, -0.1611328125, -0.8125, -3.328125, 1.0859375, 2.359375, -0.404296875, -0.5859375, -2.09375, 0.359375, 2.5625, 3.453125, 2.53125, 3.328125, -0.6015625, 1.9140625, 0.9453125, -2.96875, -3.953125, -0.4296875, 4.59375, 2.5625, -0.326171875, 2.359375, -0.458984375, 2.546875, -5.84375, 2.046875, -4.4375, -1.5390625, 3.046875, 1.734375, -1.5703125, -2.0625, -1.8359375, 0.75, -0.84765625, 0.486328125, -0.625, 0.7578125, -0.5859375, -2.640625, -1.2109375, 0.609375, 3.625, -4.59375, 2.75, -2.703125, 1.5, -0.57421875, 0.19140625, -1.6640625, -1.453125, 2.828125, 3.265625, 2.375, -1.8515625, -0.515625, -0.515625, 2.03125, 1.5, 0.90234375, -0.60546875, -0.72265625, 2.703125, -2.046875, -3.8125, -0.08056640625, -1.40625, -2.21875, -2.609375, -2.0625, -1.3046875, -0.134765625, -1.703125, -2.3125, 1.6171875, 0.1982421875, 0.95703125, 1.2265625, -1.125, -0.7734375, -0.375, -0.9609375, -4.21875, 0.42578125, 1.53125, -2.015625, 1.6875, 1.1171875, -0.6171875, 2.390625, 1.25, 0.3359375, 2.796875, 1.3203125, -1.5546875, -2.203125, 0.46484375, -0.35546875, -3.234375, -0.98828125, 0.6484375, -0.5546875, 0.4921875, -2.296875, -2.515625, 0.8671875, -0.0771484375, -0.69140625, 0.9453125, 0.52734375, 0.91796875, -1.5078125, -2.421875, -1.7890625, -1.5, 1.9375, 0.8671875, -2.265625, -1.46875, -2.765625, 2.453125, 1.078125, 3.0, 1.59375, 0.328125, -3.0625, -0.5546875, -1.296875, -1.609375, 0.3203125, -1.515625, -1.0703125, -5.65625, -0.734375, 2.34375, 0.81640625, -1.7890625, -2.125, 0.78125, 2.609375, 2.1875, 4.84375, 1.671875, 1.6484375, 0.462890625, -0.3984375, -1.8984375, -3.6875, 0.030517578125, -2.21875, -1.328125, -3.015625, 2.734375, 3.609375, 1.0234375, -1.1640625, -2.328125, -1.0625, -1.7578125, -0.6640625, 0.98046875, -1.2578125, 0.4609375, -1.9375, -0.046875, -3.25, -2.96875, -3.78125, -3.359375, -0.51171875, 2.828125, -2.828125]], "decoder": [[[-2.205371856689453e-06, 2.384185791015625e-06, 6.318092346191406e-06, 4.827976226806641e-06, -7.927417755126953e-06, -9.47713851928711e-06, 7.152557373046875e-06, 8.165836334228516e-06, -3.3080577850341797e-06, -1.3113021850585938e-06, 5.125999450683594e-06, -5.781650543212891e-06, -3.129243850708008e-06, 7.510185241699219e-06, 1.7881393432617188e-06, 3.874301910400391e-06, -5.930662155151367e-06, -5.036592483520508e-06, -1.5795230865478516e-06, 6.556510925292969e-07, 2.8014183044433594e-06, -4.470348358154297e-07, -1.6093254089355469e-06, -3.2782554626464844e-07, 2.4139881134033203e-06, 9.5367431640625e-07, 8.046627044677734e-07, -1.2516975402832031e-06, -8.940696716308594e-08, -2.5480985641479492e-06, -2.294778823852539e-06, -8.195638656616211e-07, -9.5367431640625e-07, 1.0132789611816406e-06, 1.1324882507324219e-06, 8.940696716308594e-08, -1.087784767150879e-06, -2.60770320892334e-06, -1.3113021850585938e-06, -9.238719940185547e-07, 2.980232238769531e-07, 0.0, 1.3113021850585938e-06, 6.258487701416016e-07, 9.238719940185547e-07, -4.172325134277344e-07, -3.2782554626464844e-07, 1.341104507446289e-06, 5.364418029785156e-07, -5.960464477539062e-07, -5.960464477539063e-08, -2.682209014892578e-07, 2.9802322387695312e-08, -7.301568984985352e-07, 6.556510925292969e-07, 6.854534149169922e-07, -2.384185791015625e-07, -6.854534149169922e-07, -8.046627044677734e-07, 4.470348358154297e-07, 4.172325134277344e-07, 6.556510925292969e-07, 0.0, 4.470348358154297e-07, 3.2782554626464844e-07, -8.940696716308594e-08, 8.940696716308594e-08, 5.364418029785156e-07, 6.854534149169922e-07, -1.4901161193847656e-07, -2.9802322387695312e-08, 2.980232238769531e-07, 2.682209014892578e-07, -1.4901161193847656e-07, 2.9802322387695312e-08, -3.8743019104003906e-07, -6.854534149169922e-07, -5.066394805908203e-07, -2.384185791015625e-07, 1.1920928955078125e-07, 1.1920928955078125e-07, -2.980232238769531e-07, -1.7881393432617188e-07, 3.2782554626464844e-07, 5.364418029785156e-07, 1.7881393432617188e-07, 0.0, 3.8743019104003906e-07, 0.0, -2.980232238769531e-07, 0.0, 7.450580596923828e-07, 6.258487701416016e-07, 2.9802322387695312e-08, 3.2782554626464844e-07, 5.364418029785156e-07, 3.2782554626464844e-07, -5.960464477539063e-08, -5.960464477539063e-08, 4.172325134277344e-07, 1.4901161193847656e-07, -4.172325134277344e-07, -2.9802322387695312e-08, 4.470348358154297e-07, 3.2782554626464844e-07, -5.066394805908203e-07, -8.940696716308594e-08, 4.76837158203125e-07, -5.960464477539063e-08, -4.76837158203125e-07, -3.5762786865234375e-07, 2.682209014892578e-07, 1.1920928955078125e-07, -6.258487701416016e-07, -2.384185791015625e-07, 1.1920928955078125e-06, 5.662441253662109e-07, -2.086162567138672e-07, -2.086162567138672e-07, 1.4901161193847656e-07, -8.940696716308594e-08, -8.046627044677734e-07, -4.76837158203125e-07, -2.086162567138672e-07, -8.940696716308594e-08, -8.940696716308594e-08, 1.7881393432617188e-07, 7.152557373046875e-07, -1.1920928955078125e-07, -7.450580596923828e-07, -2.086162567138672e-07, 3.5762786865234375e-07, 1.4901161193847656e-07, -1.4901161193847656e-07, 2.9802322387695312e-08, 2.682209014892578e-07, 0.0, -2.980232238769531e-07, -8.940696716308594e-08, -1.1920928955078125e-07, -2.384185791015625e-07, -8.940696716308594e-08, 2.384185791015625e-07, 4.470348358154297e-07, 2.384185791015625e-07, -3.5762786865234375e-07, -3.5762786865234375e-07, 2.980232238769531e-07, 2.384185791015625e-07, 4.76837158203125e-07, 3.8743019104003906e-07, 2.384185791015625e-07, 2.9802322387695312e-08, -1.7881393432617188e-07, 1.4901161193847656e-07, 3.8743019104003906e-07, 1.1920928955078125e-07, -8.940696716308594e-08, 0.0, 2.682209014892578e-07, 0.0, -2.980232238769531e-07, -1.1920928955078125e-07, 4.172325134277344e-07, 5.364418029785156e-07, 1.4901161193847656e-07, 3.8743019104003906e-07, 2.682209014892578e-07, 0.0, -3.8743019104003906e-07, -1.1920928955078125e-07, 2.980232238769531e-07, 2.384185791015625e-07, 2.384185791015625e-07, 2.9802322387695312e-08, 4.470348358154297e-07, 0.0, 2.9802322387695312e-08, 2.682209014892578e-07, 8.940696716308594e-08, 2.086162567138672e-07, -2.384185791015625e-07, -4.172325134277344e-07, 2.384185791015625e-07, 2.9802322387695312e-08, -4.76837158203125e-07, 0.0, 5.066394805908203e-07, 3.5762786865234375e-07, -1.7881393432617188e-07, -1.7881393432617188e-07, 3.5762786865234375e-07, 6.854534149169922e-07, 1.4901161193847656e-07, 1.7881393432617188e-07, 7.748603820800781e-07, 1.7881393432617188e-07, -5.066394805908203e-07, -1.1920928955078125e-07, 3.8743019104003906e-07]], [[-2.5033950805664062e-05, -3.743171691894531e-05, -8.20159912109375e-05, -0.00018024444580078125, -0.00029754638671875, -0.000274658203125, -0.00013637542724609375, -9.298324584960938e-06, 8.678436279296875e-05, 0.00021076202392578125, 0.00022411346435546875, 0.00012111663818359375, 3.933906555175781e-05, 2.7418136596679688e-05, 6.4849853515625e-05, 0.000171661376953125, 0.000125885009765625, -5.21540641784668e-06, -0.00018596649169921875, -0.0002040863037109375, -9.632110595703125e-05, -3.647804260253906e-05, -1.4603137969970703e-05, -2.944469451904297e-05, -1.424551010131836e-05, 8.678436279296875e-05, 0.0002956390380859375, 0.000423431396484375, 0.000331878662109375, 0.000255584716796875, 0.00011730194091796875, -0.0001201629638671875, -0.00021076202392578125, -0.0001926422119140625, -0.000301361083984375, -0.000492095947265625, -0.0005340576171875, -0.0003566741943359375, -0.000179290771484375, -8.487701416015625e-05, -3.9577484130859375e-05, -1.5854835510253906e-05, -0.00010776519775390625, -0.0002460479736328125, -0.0002803802490234375, -0.000324249267578125, -0.0002288818359375, -8.58306884765625e-05, -0.00012683868408203125, -0.0001430511474609375, -3.62396240234375e-05, -3.4332275390625e-05, -0.000156402587890625, -0.000240325927734375, -0.00031280517578125, -0.000335693359375, -0.0003986358642578125, -0.0004711151123046875, -0.0003108978271484375, -7.915496826171875e-05, 0.00010776519775390625, 3.600120544433594e-05, -0.00014781951904296875, -0.0004138946533203125, -0.000583648681640625, -0.00048065185546875, -0.000308990478515625, -0.000316619873046875, -0.0003147125244140625, -0.0004024505615234375, -0.000629425048828125, -0.000591278076171875, -0.0002899169921875, 0.00012969970703125, 0.0004177093505859375, 0.0003986358642578125, 0.0003299713134765625, 0.0002918243408203125, 0.000202178955078125, 0.0001010894775390625, -3.5762786865234375e-05, -0.000179290771484375, -0.00023937225341796875, -6.580352783203125e-05, 7.82012939453125e-05, -0.00011539459228515625, -0.00043487548828125, -0.000591278076171875, -0.0002574920654296875, 0.0003185272216796875, 0.000537872314453125, 0.000457763671875, 0.000335693359375, 9.250640869140625e-05, -0.0001735687255859375, -0.0002536773681640625, -0.0003643035888671875, -0.00048065185546875, -0.0003795623779296875, -0.000370025634765625, -0.0004596710205078125, -0.000507354736328125, -0.00034332275390625, -5.340576171875e-05, 0.000274658203125, 0.000438690185546875, 0.00034332275390625, 9.34600830078125e-05, -3.4570693969726562e-06, -2.771615982055664e-06, -3.0994415283203125e-05, -6.031990051269531e-05, -0.00018596649169921875, -0.0003509521484375, -0.0003032684326171875, -0.0001277923583984375, 5.4836273193359375e-05, 0.0002651214599609375, 0.000492095947265625, 0.000591278076171875, 0.00058746337890625, 0.000614166259765625, 0.00061798095703125, 0.00049591064453125, 0.0002574920654296875, -9.98377799987793e-07, -0.00019550323486328125, -0.000263214111328125, -0.00013065338134765625, 0.00017452239990234375, 0.0003719329833984375, 0.0004119873046875, 0.0002841949462890625, 0.00022602081298828125, 2.9325485229492188e-05, -3.933906555175781e-05, 0.000110626220703125, 0.00014972686767578125, 4.00543212890625e-05, -0.00015735626220703125, -0.0003452301025390625, -0.0003871917724609375, -7.724761962890625e-05, 0.00040435791015625, 0.000469207763671875, 0.0001983642578125, 5.316734313964844e-05, -0.00012493133544921875, -0.00030517578125, -0.0004730224609375, -0.0004482269287109375, -0.0004520416259765625, -0.0006256103515625, -0.000720977783203125, -0.000812530517578125, -0.000713348388671875, -0.0004673004150390625, -0.00011730194091796875, 0.00016880035400390625, 0.000324249267578125, 0.0002498626708984375, 0.0002498626708984375, 0.000331878662109375, 0.0001544952392578125, -0.000270843505859375, -0.000690460205078125, -0.000713348388671875, -0.000461578369140625, -0.0002956390380859375, -0.00018978118896484375, -3.647804260253906e-05, 0.000118255615234375, 0.000232696533203125, 0.0002918243408203125, 0.0002803802490234375, 0.00017833709716796875, 1.1920928955078125e-05, -0.000240325927734375, -0.0002994537353515625, -0.00020885467529296875, -0.0002899169921875, -0.000446319580078125, -0.0004444122314453125, -0.0005035400390625, -0.000484466552734375, -0.0002574920654296875, -2.9802322387695312e-05, 0.00017070770263671875, 0.00016117095947265625, 4.291534423828125e-05, -0.0001430511474609375, -0.0003147125244140625, -0.00019741058349609375, 9.1552734375e-05, 0.0002002716064453125, 0.0001163482666015625, 5.888938903808594e-05, -9.822845458984375e-05, -0.0001354217529296875, -4.1961669921875e-05]]]},
+    "xpu": {"encoder": [[2.15625, 3.03125, -1.4140625, 5.03125, -6.1875, -11.125, -0.03466796875, 0.69921875, -0.294921875, 2.625, 3.015625, 4.90625, 1.421875, -5.28125, -0.5, 2.203125, -5.625, 1.3359375, 5.3125, 4.5, -1.015625, 15.625, -3.6875, 2.328125, 1.8515625, 3.453125, 0.458984375, 1.484375, 0.380859375, -4.3125, 2.84375, -0.3828125, 2.828125, 2.859375, -2.09375, -5.125, -2.390625, -4.15625, 3.96875, 10.375, 1.0078125, 7.9375, 12.5, -3.421875, 0.5546875, 0.32421875, -1.9609375, -7.1875, 1.546875, -6.40625, 3.15625, 0.1171875, -0.9375, 4.1875, 2.203125, -11.4375, -2.0, -4.4375, 1.4453125, -5.875, -2.21875, 2.515625, 5.71875, 0.271484375, -0.59375, -3.171875, -15.1875, -2.546875, 7.53125, -2.46875, -0.298828125, 2.828125, 8.4375, 0.703125, 3.5, 0.486328125, 2.21875, 0.71875, 4.9375, -3.296875, 2.546875, 4.375, 1.7265625, 11.9375, -4.6875, 5.09375, -1.9453125, -9.875, 4.21875, 0.9140625, -9.375, 1.796875, 2.28125, -6.65625, 1.90625, -1.6171875, 3.21875, 12.1875, -1.4375, -0.5234375, -4.875, -1.328125, -1.3125, -9.25, -0.3515625, -5.6875, 0.240234375, 6.375, 5.625, 3.03125, 6.25, 8.625, -3.0625, -7.40625, 5.0625, -4.5, -13.3125, -1.6640625, 3.421875, 2.875, -0.5546875, 4.3125, 2.453125, -3.609375, -8.1875, -1.2734375, 4.875, 1.234375, 10.75, -0.5625, -5.40625, -3.3125, 3.28125, 16.125, 10.9375, -8.25, 8.625, 5.40625, -0.51953125, -8.375, 10.0625, 6.6875, 1.9140625, -0.64453125, -10.875, -0.07568359375, 5.71875, -3.75, 6.6875, -0.8984375, 14.0625, -5.96875, -8.5, -6.90625, 7.96875, -4.75, -3.921875, -6.4375, 8.125, -0.13671875, -3.28125, 8.25, 7.71875, 0.46484375, -10.125, -2.953125, 2.71875, -4.4375, 6.34375, 14.8125, -9.5, -18.25, 11.3125, 4.9375, -2.375, 0.71484375, -1.2265625, 7.53125, -2.84375, -4.5, -2.390625, 5.03125, -0.82421875, -6.75, 7.4375, -4.375, 0.380859375, 6.15625, -5.71875, 5.8125, -4.71875, 11.875, -7.5625, 3.765625, -2.078125, 3.984375, 5.375, 7.09375, 4.5625, 3.140625], [1.4609375, 1.0546875, -2.796875, 3.53125, -4.78125, -2.5, 1.1953125, 1.09375, 4.8125, 0.99609375, -0.1376953125, -2.90625, 2.140625, 1.046875, -1.3125, -0.1904296875, 5.34375, -1.3203125, -0.37109375, -2.28125, 0.388671875, 0.408203125, 3.34375, 8.1875, -0.9609375, -4.65625, -2.640625, -0.103515625, -0.7578125, -3.375, 1.171875, 2.359375, -0.34375, -0.66796875, -2.015625, 0.310546875, 2.5625, 3.53125, 2.53125, 3.34375, -0.59765625, 1.90625, 0.9765625, -2.859375, -3.953125, -0.486328125, 4.625, 2.515625, -0.45703125, 2.359375, -0.4609375, 2.53125, -5.8125, 2.0, -4.46875, -1.578125, 3.0625, 1.6953125, -1.6171875, -2.125, -1.84375, 0.77734375, -0.85546875, 0.51953125, -0.75, 0.77734375, -0.6328125, -2.65625, -1.171875, 0.44921875, 3.46875, -4.5625, 2.75, -2.640625, 1.5078125, -0.66796875, 0.099609375, -1.578125, -1.3984375, 2.875, 3.109375, 2.359375, -2.015625, -0.44140625, -0.42578125, 2.078125, 1.4921875, 0.890625, -0.58984375, -0.73828125, 2.609375, -2.140625, -3.875, 0.06103515625, -1.546875, -2.109375, -2.609375, -1.8671875, -1.265625, -0.181640625, -1.609375, -2.328125, 1.546875, 0.3671875, 1.0546875, 1.2109375, -1.015625, -0.88671875, -0.4453125, -0.9921875, -4.25, 0.5, 1.2734375, -1.9375, 1.671875, 1.15625, -0.482421875, 2.4375, 1.21875, 0.498046875, 2.828125, 1.3515625, -1.5703125, -2.265625, 0.53515625, -0.427734375, -3.140625, -0.9765625, 0.78515625, -0.65625, 0.69140625, -2.203125, -2.546875, 0.8671875, -0.004638671875, -0.63671875, 0.90234375, 0.466796875, 0.87109375, -1.4140625, -2.40625, -2.0, -1.6328125, 1.90625, 0.984375, -2.203125, -1.65625, -2.625, 2.671875, 1.3125, 2.9375, 1.703125, 0.439453125, -2.90625, -0.419921875, -1.5, -1.6953125, 0.44140625, -1.609375, -1.1328125, -5.6875, -0.83984375, 2.265625, 0.890625, -1.6875, -2.171875, 0.875, 2.609375, 1.953125, 4.875, 1.6953125, 1.640625, 0.34765625, -0.7890625, -1.9375, -3.53125, -0.06884765625, -2.5, -1.3671875, -3.046875, 2.921875, 3.6875, 1.0625, -1.140625, -2.328125, -1.0859375, -1.640625, -0.71875, 1.1015625, -1.2734375, 0.62109375, -1.875, -0.1083984375, -3.234375, -2.984375, -3.84375, -3.3125, -0.4140625, 2.796875, -2.859375]], "decoder": [[[-2.3543834686279297e-06, -6.28642737865448e-08, 1.6540288925170898e-06, 6.467103958129883e-06, 1.4007091522216797e-06, -4.559755325317383e-06, 7.68899917602539e-06, 7.748603820800781e-06, -3.129243850708008e-06, -2.175569534301758e-06, 3.933906555175781e-06, -5.424022674560547e-06, -3.904104232788086e-06, 6.765127182006836e-06, 1.169741153717041e-06, 3.9637088775634766e-06, -5.8710575103759766e-06, -4.649162292480469e-06, -2.9206275939941406e-06, 3.073364496231079e-07, 2.950429916381836e-06, -6.183981895446777e-07, -1.2740492820739746e-06, -3.3527612686157227e-07, 2.8759241104125977e-06, 1.3262033462524414e-06, 8.009374141693115e-07, -1.5124678611755371e-06, -1.9907020032405853e-08, -2.2351741790771484e-06, -1.3634562492370605e-06, -2.2724270820617676e-07, -7.711350917816162e-07, 7.227063179016113e-07, 3.259629011154175e-07, -5.289912223815918e-07, -1.30385160446167e-06, -2.7418136596679688e-06, -1.2218952178955078e-06, -1.0207295417785645e-06, -1.076841726899147e-08, -4.33996319770813e-07, 1.125037670135498e-06, 4.5634806156158447e-07, 6.705522537231445e-07, -4.3585896492004395e-07, -4.2654573917388916e-07, 1.0728836059570312e-06, -5.675246939063072e-09, -4.2654573917388916e-07, -7.171183824539185e-08, -1.0849907994270325e-07, 5.774199962615967e-07, -6.221234798431396e-07, 6.854534149169922e-07, 7.078051567077637e-07, -3.0919909477233887e-07, -5.699694156646729e-07, -5.960464477539062e-07, 5.066394805908203e-07, 5.885958671569824e-07, 3.594905138015747e-07, -6.705522537231445e-07, -1.4808028936386108e-07, -1.5832483768463135e-07, 1.5273690223693848e-07, 1.0523945093154907e-07, 2.3562461137771606e-07, 3.9301812648773193e-07, -1.4062970876693726e-07, 9.837094694375992e-09, 2.0302832126617432e-07, 7.729977369308472e-08, -2.644956111907959e-07, -3.4226104617118835e-08, -3.073364496231079e-07, -6.556510925292969e-07, -6.817281246185303e-07, -5.327165126800537e-07, -3.189779818058014e-08, -2.337619662284851e-07, -1.1920928955078125e-07, -8.475035429000854e-08, 1.210719347000122e-07, 4.3213367462158203e-07, -2.421438694000244e-07, -1.0384246706962585e-07, 1.1175870895385742e-07, -2.3655593395233154e-07, -3.632158041000366e-07, 3.306195139884949e-08, 9.760260581970215e-07, 4.3213367462158203e-07, -1.2014061212539673e-07, 1.30385160446167e-07, 6.742775440216064e-07, 1.51805579662323e-07, -4.842877388000488e-07, -7.152557373046875e-07, 8.381903171539307e-08, -2.3166649043560028e-08, -4.414469003677368e-07, 6.184563972055912e-10, 2.2631138563156128e-07, 3.8743019104003906e-07, -5.289912223815918e-07, -3.725290298461914e-07, 4.2282044887542725e-07, 1.4901161193847656e-07, -4.302710294723511e-07, -2.8312206268310547e-07, 2.868473529815674e-07, -7.869675755500793e-08, -1.125037670135498e-06, -5.289912223815918e-07, 8.270144462585449e-07, 5.774199962615967e-07, -1.1222437024116516e-07, -2.5890767574310303e-07, 1.7601996660232544e-07, 1.4726538211107254e-08, -4.917383193969727e-07, -4.0046870708465576e-07, -2.337619662284851e-07, -1.8742866814136505e-08, -1.4901161193847656e-07, 6.798654794692993e-08, 6.780028343200684e-07, -3.91155481338501e-07, -9.499490261077881e-07, -4.0978193283081055e-07, 3.0174851417541504e-07, 8.42846930027008e-08, -1.150183379650116e-07, 2.421438694000244e-07, 2.2351741790771484e-07, -8.335337042808533e-08, -4.842877388000488e-07, -2.3562461137771606e-07, 1.1408701539039612e-07, -1.594889909029007e-08, 1.1548399925231934e-07, 2.1513551473617554e-07, 3.110617399215698e-07, 2.7567148208618164e-07, -5.699694156646729e-07, -4.6938657760620117e-07, 8.754432201385498e-08, 4.470348358154297e-08, 2.421438694000244e-07, 4.041939973831177e-07, 6.183981895446777e-07, 6.891787052154541e-07, -8.754432201385498e-08, -7.683411240577698e-08, 3.557652235031128e-07, -2.9569491744041443e-08, -5.820766091346741e-08, 9.406358003616333e-08, 2.6263296604156494e-07, 1.9650906324386597e-07, -5.51808625459671e-08, 6.426125764846802e-08, 6.742775440216064e-07, 6.034970283508301e-07, 1.8044374883174896e-08, 1.4901161193847656e-07, 2.253800630569458e-07, 1.94646418094635e-07, -5.178153514862061e-07, -5.21540641784668e-07, -4.0745362639427185e-08, 1.019798219203949e-07, -1.2014061212539673e-07, 3.3032847568392754e-09, 6.780028343200684e-07, 1.1455267667770386e-07, -1.8440186977386475e-07, -7.59027898311615e-08, -1.7974525690078735e-07, 1.4493707567453384e-08, 8.840288501232862e-10, -1.862645149230957e-07, 2.5890767574310303e-07, -1.778826117515564e-07, -3.520399332046509e-07, 6.798654794692993e-08, 4.991888999938965e-07, 4.3958425521850586e-07, -2.7008354663848877e-07, -3.203749656677246e-07, 1.6205012798309326e-07, 2.477318048477173e-07, -3.594905138015747e-07, -1.7974525690078735e-07, 5.364418029785156e-07, 4.805624485015869e-07, -1.4621764421463013e-07, -1.685693860054016e-07, 4.954636096954346e-07]], [[-2.682209014892578e-05, -4.124641418457031e-05, -7.677078247070312e-05, -0.000171661376953125, -0.000293731689453125, -0.000270843505859375, -0.000125885009765625, 1.0654330253601074e-06, 8.678436279296875e-05, 0.00020885467529296875, 0.00022029876708984375, 0.0001239776611328125, 4.291534423828125e-05, 2.86102294921875e-05, 6.198883056640625e-05, 0.0001697540283203125, 0.00012493133544921875, -7.092952728271484e-06, -0.0001888275146484375, -0.0002002716064453125, -9.1552734375e-05, -3.457069396972656e-05, -1.537799835205078e-05, -3.266334533691406e-05, -1.633167266845703e-05, 8.392333984375e-05, 0.0002918243408203125, 0.0004138946533203125, 0.000324249267578125, 0.0002536773681640625, 0.00012063980102539062, -0.00011301040649414062, -0.00020599365234375, -0.00018787384033203125, -0.0002994537353515625, -0.000484466552734375, -0.000522613525390625, -0.0003452301025390625, -0.0001659393310546875, -8.0108642578125e-05, -3.933906555175781e-05, -1.8715858459472656e-05, -0.00010824203491210938, -0.0002460479736328125, -0.0002841949462890625, -0.0003204345703125, -0.0002231597900390625, -7.581710815429688e-05, -0.0001087188720703125, -0.00013065338134765625, -2.086162567138672e-05, -1.7404556274414062e-05, -0.00013828277587890625, -0.00022792816162109375, -0.000308990478515625, -0.000335693359375, -0.000392913818359375, -0.000453948974609375, -0.0003032684326171875, -8.678436279296875e-05, 9.822845458984375e-05, 3.4332275390625e-05, -0.0001373291015625, -0.000392913818359375, -0.000560760498046875, -0.00046539306640625, -0.0003070831298828125, -0.0003204345703125, -0.0003223419189453125, -0.000400543212890625, -0.000621795654296875, -0.000583648681640625, -0.0002956390380859375, 0.00011301040649414062, 0.0003910064697265625, 0.0003814697265625, 0.0003261566162109375, 0.0002880096435546875, 0.00019931793212890625, 8.869171142578125e-05, -4.9114227294921875e-05, -0.0001811981201171875, -0.00023937225341796875, -6.723403930664062e-05, 8.249282836914062e-05, -0.00011444091796875, -0.0004405975341796875, -0.000598907470703125, -0.0002727508544921875, 0.0003032684326171875, 0.00052642822265625, 0.00045013427734375, 0.0003261566162109375, 8.58306884765625e-05, -0.00018024444580078125, -0.0002651214599609375, -0.00037384033203125, -0.0004825592041015625, -0.0003795623779296875, -0.000370025634765625, -0.000461578369140625, -0.000507354736328125, -0.0003414154052734375, -4.9591064453125e-05, 0.0002727508544921875, 0.0004367828369140625, 0.0003452301025390625, 9.5367431640625e-05, -2.3692846298217773e-06, -2.086162567138672e-06, -3.170967102050781e-05, -6.246566772460938e-05, -0.00018405914306640625, -0.000347137451171875, -0.000301361083984375, -0.0001220703125, 6.532669067382812e-05, 0.000270843505859375, 0.0004863739013671875, 0.000583648681640625, 0.000579833984375, 0.000606536865234375, 0.000614166259765625, 0.00048828125, 0.00024318695068359375, -1.055002212524414e-05, -0.00019741058349609375, -0.0002593994140625, -0.0001239776611328125, 0.0001811981201171875, 0.0003795623779296875, 0.00041961669921875, 0.000293731689453125, 0.00023174285888671875, 2.8252601623535156e-05, -4.696846008300781e-05, 9.5367431640625e-05, 0.00013637542724609375, 2.9921531677246094e-05, -0.0001583099365234375, -0.0003376007080078125, -0.000377655029296875, -7.152557373046875e-05, 0.00040435791015625, 0.00046539306640625, 0.000202178955078125, 5.817413330078125e-05, -0.00012874603271484375, -0.000308990478515625, -0.0004825592041015625, -0.0004558563232421875, -0.000461578369140625, -0.000637054443359375, -0.000728607177734375, -0.00081634521484375, -0.000705718994140625, -0.0004558563232421875, -0.00010585784912109375, 0.00017642974853515625, 0.0003261566162109375, 0.0002536773681640625, 0.0002460479736328125, 0.0003185272216796875, 0.00014019012451171875, -0.0002880096435546875, -0.000701904296875, -0.000720977783203125, -0.0004596710205078125, -0.000286102294921875, -0.00017547607421875, -2.467632293701172e-05, 0.00011920928955078125, 0.0002307891845703125, 0.000278472900390625, 0.000255584716796875, 0.00014400482177734375, -2.276897430419922e-05, -0.00026702880859375, -0.000324249267578125, -0.000225067138671875, -0.0002994537353515625, -0.00045013427734375, -0.0004444122314453125, -0.000499725341796875, -0.0004749298095703125, -0.00025177001953125, -3.2901763916015625e-05, 0.00016307830810546875, 0.0001544952392578125, 3.5762786865234375e-05, -0.00015163421630859375, -0.0003185272216796875, -0.00019931793212890625, 8.916854858398438e-05, 0.00019931793212890625, 0.00011444091796875, 5.650520324707031e-05, -9.679794311523438e-05, -0.00012969970703125, -4.100799560546875e-05]]]}
+}
\ No newline at end of file

filename: tests/models/vibevoice_acoustic_tokenizer/test_modeling_vibevoice_acoustic_tokenizer.py
patch: @@ -310,8 +310,12 @@ def test_batch_integration(self):
         )
         with open(RESULTS_PATH, "r") as f:
             expected_results = json.load(f)
-        expected_encoder = torch.tensor(expected_results["encoder"]).to(dtype)
-        expected_decoder = torch.tensor(expected_results["decoder"]).to(dtype)
+
+        # Get device-specific expected results
+        device_key = torch_device if torch_device in expected_results else "cuda"
+        device_results = expected_results[device_key]
+        expected_encoder = torch.tensor(device_results["encoder"]).to(dtype)
+        expected_decoder = torch.tensor(device_results["decoder"]).to(dtype)
 
         # Prepare inputs
         audio_paths = [