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;