Skip to content

Commit a825c5a

Browse files
authored
Use pytorch for CPU (#645)
* Use pytorch for CPU * Update onnx_ob.js
1 parent ec4ab51 commit a825c5a

File tree

2 files changed

+12
-5
lines changed

2 files changed

+12
-5
lines changed

lib/model/nns/onnx/onnx_pb.js

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,11 @@
11
var window = typeof window !== 'undefined' ? window : null; var self = typeof self !== 'undefined' ? self : null;
22
var commonjsGlobal = typeof globalThis !== 'undefined' ? globalThis : typeof window !== 'undefined' ? window : typeof global !== 'undefined' ? global : typeof self !== 'undefined' ? self : {};
33

4-
var onnx_pb = {};
4+
function getDefaultExportFromCjs (x) {
5+
return x && x.__esModule && Object.prototype.hasOwnProperty.call(x, 'default') ? x['default'] : x;
6+
}
7+
8+
var onnx_pb$1 = {};
59

610
var googleProtobuf = {};
711

@@ -539,7 +543,7 @@ goog.TRUSTED_TYPES_POLICY_NAME="";goog.identity_=function(a){return a};goog.crea
539543
jspb.Message.prototype.cloneMessage=function(){return jspb.Message.cloneMessage(this)};goog.exportProperty(jspb.Message.prototype,"cloneMessage",jspb.Message.prototype.cloneMessage);jspb.Message.prototype.clone=function(){return jspb.Message.cloneMessage(this)};goog.exportProperty(jspb.Message.prototype,"clone",jspb.Message.prototype.clone);jspb.Message.clone=function(a){return jspb.Message.cloneMessage(a)};goog.exportProperty(jspb.Message,"clone",jspb.Message.clone);jspb.Message.cloneMessage=function(a){return new a.constructor(jspb.Message.clone_(a.toArray()))};
540544
jspb.Message.copyInto=function(a,b){jspb.asserts.assertInstanceof(a,jspb.Message);jspb.asserts.assertInstanceof(b,jspb.Message);jspb.asserts.assert(a.constructor==b.constructor,"Copy source and target message should have the same type.");a=jspb.Message.clone(a);for(var c=b.toArray(),d=a.toArray(),e=c.length=0;e<d.length;e++)c[e]=d[e];b.wrappers_=a.wrappers_;b.extensionObject_=a.extensionObject_;};goog.exportProperty(jspb.Message,"copyInto",jspb.Message.copyInto);
541545
jspb.Message.clone_=function(a){if(Array.isArray(a)){for(var b=Array(a.length),c=0;c<a.length;c++){var d=a[c];null!=d&&(b[c]="object"==typeof d?jspb.Message.clone_(jspb.asserts.assert(d)):d);}return b}if(jspb.Message.SUPPORTS_UINT8ARRAY_&&a instanceof Uint8Array)return new Uint8Array(a);b={};for(c in a)d=a[c],null!=d&&(b[c]="object"==typeof d?jspb.Message.clone_(jspb.asserts.assert(d)):d);return b};jspb.Message.registerMessageType=function(a,b){b.messageId=a;};
542-
goog.exportProperty(jspb.Message,"registerMessageType",jspb.Message.registerMessageType);jspb.Message.messageSetExtensions={};jspb.Message.messageSetExtensionsBinary={};jspb.Export={};(exports.Map=jspb.Map,exports.Message=jspb.Message,exports.BinaryReader=jspb.BinaryReader,exports.BinaryWriter=jspb.BinaryWriter,exports.ExtensionFieldInfo=jspb.ExtensionFieldInfo,exports.ExtensionFieldBinaryInfo=jspb.ExtensionFieldBinaryInfo,exports.exportSymbol=goog.exportSymbol,exports.inherits=goog.inherits,exports.object={extend:goog.object.extend},exports.typeOf=goog.typeOf);
546+
goog.exportProperty(jspb.Message,"registerMessageType",jspb.Message.registerMessageType);jspb.Message.messageSetExtensions={};jspb.Message.messageSetExtensionsBinary={};jspb.Export={};(exports.Map=jspb.Map,exports.Message=jspb.Message,exports.BinaryReader=jspb.BinaryReader,exports.BinaryWriter=jspb.BinaryWriter,exports.ExtensionFieldInfo=jspb.ExtensionFieldInfo,exports.ExtensionFieldBinaryInfo=jspb.ExtensionFieldBinaryInfo,exports.exportSymbol=goog.exportSymbol,exports.inherits=goog.inherits,exports.object={extend:goog.object.extend},exports.typeOf=goog.typeOf);
543547
} (googleProtobuf));
544548

545549
(function (exports) {
@@ -8581,7 +8585,9 @@ goog.TRUSTED_TYPES_POLICY_NAME="";goog.identity_=function(a){return a};goog.crea
85818585
STABLE: 1
85828586
};
85838587

8584-
goog.object.extend(exports, proto.onnx);
8585-
} (onnx_pb));
8588+
goog.object.extend(exports, proto.onnx);
8589+
} (onnx_pb$1));
8590+
8591+
var onnx_pb = /*@__PURE__*/getDefaultExportFromCjs(onnx_pb$1);
85868592

85878593
export { onnx_pb as default };

onnx.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,8 @@ services:
116116
- -c
117117
- >
118118
pip install --upgrade pip
119-
&& pip install black numpy torch onnx
119+
&& pip install black numpy onnx
120+
&& pip install torch --index-url https://download.pytorch.org/whl/cpu
120121
&& black /root_dir/tests/lib/model/nns/onnx
121122
&& python /app/create_onnx.py
122123
EOS

0 commit comments

Comments
 (0)