Update Face plugin to incorporate TensorFlow GPU handling update
parent
474a3c599f
commit
49eb779f80
|
@ -0,0 +1,2 @@
|
|||
TF_FORCE_GPU_ALLOW_GROWTH=true
|
||||
#CUDA_VISIBLE_DEVICES=0,2
|
|
@ -10,7 +10,7 @@
|
|||
"socket.io": "^2.0.4",
|
||||
"face-api.js": "^0.22.2",
|
||||
"canvas": "^2.1.0",
|
||||
"@tensorflow/tfjs-node-gpu": "^0.1.21"
|
||||
"dotenv": "^8.2.0"
|
||||
},
|
||||
"devDependencies": {},
|
||||
"scripts": {
|
||||
|
|
|
@ -10,6 +10,7 @@
|
|||
// Base Init >>
|
||||
var fs = require('fs');
|
||||
var config = require('./conf.json')
|
||||
var dotenv = require('dotenv').config()
|
||||
var s
|
||||
try{
|
||||
s = require('../pluginBase.js')(__dirname,config)
|
||||
|
|
Loading…
Reference in New Issue