From 2b5ea7d9cb5d48c9286a4bf2e6a20bfb9e960428 Mon Sep 17 00:00:00 2001
From: Ahmed Harmouche <ahmedharmouche92@gmail.com>
Date: Wed, 18 Oct 2023 00:28:19 +0200
Subject: [PATCH] Fix output Float32Array size in webgpu export (#2096)

---
 extra/export_model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/extra/export_model.py b/extra/export_model.py
index 70a7fe48a3..fe90535ef3 100644
--- a/extra/export_model.py
+++ b/extra/export_model.py
@@ -121,7 +121,7 @@ const setupNet = async (device, safetensor) => {{
         device.queue.submit([gpuCommands]);
 
         await gpuReadBuffer.mapAsync(GPUMapMode.READ);
-        const resultBuffer = new Float32Array(gpuReadBuffer.size);
+        const resultBuffer = new Float32Array(gpuReadBuffer.size/4);
         resultBuffer.set(new Float32Array(gpuReadBuffer.getMappedRange()));
         gpuReadBuffer.unmap();
         return resultBuffer;