Update coreml.js

lutzroeder · Jul 24, 2024 · b49faab · b49faab
1 parent 3c2b5df
commit b49faab
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/source/coreml.js b/source/coreml.js
@@ -1303,6 +1303,7 @@ coreml.Context.Graph = class {
                                     break;
                                 }
                                 case 'float16':
+                                case 'int4':
                                 case 'int8':
                                 case 'uint8': {
                                     data = stream.read(size);
@@ -1329,15 +1330,18 @@ coreml.Context.Graph = class {
                 operation.attributes[key] = convertValue(value);
             }
             operation.inputs = Object.entries(op.inputs).map(([name, input]) => {
-                const args = input.arguments.map((argument) => {
+                const value = input.arguments.map((argument) => {
+                    if (argument.value && argument.value.value && argument.value.blobFileValue) {
+                        return { name: '', value: convertValue(argument.value) };
+                    }
                     if (argument.name) {
                         const value = this.input(argument.name);
                         value.to.push(operation);
                         return value;
                     }
                     return { value: argument.value };
                 });
-                return { name, value: args };
+                return { name, value };
             });
             operation.outputs = op.outputs.map((output) => {
                 const value = this.input(output.name);
@@ -1397,6 +1401,10 @@ coreml.Context.Graph = class {
             if (value.value instanceof coreml.Tensor) {
                 value.initializer = value.value;
                 delete value.value;
+                if (name === '') {
+                    this.values.set(value, value);
+                    return value;
+                }
             }
             if (!this.values.has(name)) {
                 this.values.set(name, value);