-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathstart.js
52 lines (52 loc) · 1.19 KB
/
start.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
const os = require('os')
module.exports = async (kernel) => {
let env = {
"HF_HOME": "./huggingface",
}
if (os.platform() === 'darwin') {
env.PYTORCH_ENABLE_MPS_FALLBACK = "1",
env.PYTORCH_MPS_HIGH_WATERMARK_RATIO = "0.0"
}
return {
"run": [{
"method": "shell.start",
"params": {
"path": "app",
"env": env
}
}, {
"method": "shell.enter",
"params": {
"message": "{{os.platform() === 'win32' ? 'env\\\\Scripts\\\\activate' : 'source env/bin/activate'}} env",
"on": [{
"event": null,
"return": true
}]
}
}, {
"method": "shell.enter",
"params": {
"message": "{{os.platform() === 'win32' ? 'python' : 'python3'}} infer-web.py",
"on": [{
"event": "/(http:\/\/[0-9.:]+)/",
"return": "{{event.matches[0][1]}}"
}]
}
}, {
"method": "self.set",
"params": {
"session.json": {
"url": "{{input}}"
}
}
}, {
"method": "browser.open",
"params": {
"uri": "{{self.session.url}}",
"target": "_blank"
}
}, {
"method": "process.wait"
}]
}
}