Jelajahi Sumber

Update safetensors.js

Lutz Roeder 1 bulan lalu
induk
melakukan
b6b9f70c3b
1 mengubah file dengan 1 tambahan dan 0 penghapusan
  1. 1 0
      source/safetensors.js

+ 1 - 0
source/safetensors.js

@@ -186,6 +186,7 @@ safetensors.Reader = class {
         const stream = context.stream;
         if (stream.length > 9) {
             const buffer = stream.peek(9);
+            // Safetensors implementation caps headers length at 100 MB.
             if (buffer[4] === 0 && buffer[5] === 0 && buffer[6] === 0 && buffer[7] === 0 && buffer[8] === 0x7b) {
                 const size = (buffer[0] | buffer[1] << 8 | buffer[2] << 16 | buffer[3] << 24) >>> 0;
                 if (size < stream.length) {