diff --git a/tfjs-node-gpu/package.json b/tfjs-node-gpu/package.json index 6214674e4c0..2868e80f85f 100644 --- a/tfjs-node-gpu/package.json +++ b/tfjs-node-gpu/package.json @@ -55,6 +55,7 @@ }, "dependencies": { "@tensorflow/tfjs": "1.5.2", + "@tensorflow/tfjs-core":"1.5.2", "adm-zip": "^0.4.11", "google-protobuf": "^3.9.2", "https-proxy-agent": "^2.2.1", diff --git a/tfjs-node/package.json b/tfjs-node/package.json index 21f6e96ab05..9f60fea51ec 100644 --- a/tfjs-node/package.json +++ b/tfjs-node/package.json @@ -52,6 +52,7 @@ }, "dependencies": { "@tensorflow/tfjs": "1.5.2", + "@tensorflow/tfjs-core":"1.5.2", "adm-zip": "^0.4.11", "google-protobuf": "^3.9.2", "https-proxy-agent": "^2.2.1",