pinokiod 7.3.1 → 7.3.4

package/kernel/gpu/nvidia.js

@@ -0,0 +1,8 @@
+// NVIDIA routes to CUDA PyTorch wheels on supported non-macOS platforms.
+const supports_torch_backend = (controller, platform) => {
+  return !!controller && platform !== "darwin"
+}
+
+module.exports = {
+  supports_torch_backend
+}

package/kernel/index.js

@@ -56,7 +56,7 @@ const VARS = {
 
 //const memwatch = require('@airbnb/node-memwatch');
 class Kernel {
-  schema = "<=7.0.0"
+  schema = "<=7.4.0"
   constructor(store) {
     this.fetch = fetch
 
@@ -89,6 +89,7 @@ class Kernel {
     this.pinokio_configs = {}
     this.shellpath = shellPath.sync()
     this.favicon = new Favicon()
+    this.torch_backend = "cpu"
     this.vram = 0
     this.ram = 0
     this.sysReady = new Promise((resolve) => {
@@ -385,6 +386,9 @@ class Kernel {
   path(...args) {
     return path.resolve(this.homedir, ...args)
   }
+  systemPath(...args) {
+    return path.resolve(__dirname, "..", "system", ...args)
+  }
   exists(...args) {
     if (args) {
       let abspath = this.path(...args)
@@ -1005,6 +1009,7 @@ class Kernel {
       args: {},
     }
     this.procs = {}
+    this.activeProcessWaits = {}
     this.template = new Template()
     try {
       if (this.homedir) {
@@ -1014,6 +1019,7 @@ class Kernel {
           system,
           platform: this.platform,
           arch: this.arch,
+          torch_backend: this.torch_backend,
           vram: this.vram,
           ram: this.ram,
           proxy: (port) => {
@@ -1051,6 +1057,9 @@ class Kernel {
 
         // 2. mkdir all the folders if not already created
         await Environment.init_folders(this.homedir, this)
+        await Environment.ensurePinokioCacheDirs(this, {
+          throwOnFailure: true
+        })
 
         // if key.json doesn't exist, create an empty json file
         let ee = await this.exists(this.homedir, "key.json")
@@ -1239,6 +1248,7 @@ class Kernel {
         this.sysinfo = info
         this.gpu = info.gpu
         this.gpu_model = info.gpu_model
+        this.torch_backend = info.torch_backend || "cpu"
         this.gpus = info.gpus
         this.vram = typeof info.vram === "number" ? info.vram : 0
         this.ram = typeof info.ram === "number" ? info.ram : 0