index.js

const os = require('os')
const fs = require('fs')
const path = require('path')
class Automatic {
  // set up COMMANDLINE_ARGS inside webui-user.sh or webui.bat
  async config(req, ondata, kernel) {
    let graphics = await kernel.system.graphics()
    let platform = os.platform()
    let vendor = graphics.controllers[0].vendor
    ondata({ raw: `\r\nVendor: ${vendor}\r\n` })

    let legacy = req.params && req.params.legacy
    // if legacy => ""
    // if SDXL (not legacy) => "--no-download-sd-model"
    let defaultArgs = (legacy ? "" : "--no-download-sd-model ")
    if (platform === 'darwin') {
      if (/apple/i.test(vendor)) {
        defaultArgs += "--skip-torch-cuda-test --upcast-sampling --use-cpu interrogate --no-half --api"
      } else {
        defaultArgs += "--skip-torch-cuda-test --upcast-sampling --use-cpu all --no-half --api"
      }
      let text = await fs.promises.readFile(path.resolve(__dirname, "automatic", "webui-user.sh"), "utf8")
      let re = /^(#?)(export COMMANDLINE_ARGS=)(.+)$/m
      let newtext = text.replace(re, `$2"${defaultArgs}"`)
      await fs.promises.writeFile(path.resolve(__dirname, "automatic", "webui-user.sh"), newtext)
    } else if (platform === 'win32') {
      defaultArgs += "--xformers --no-half-vae --api"
      let text = await fs.promises.readFile(path.resolve(__dirname, "automatic", "webui.bat"), "utf8")
      let re = /^(set COMMANDLINE_ARGS=)(.*)$/m
      let newtext = text.replace(re, `$1${defaultArgs}`)
      await fs.promises.writeFile(path.resolve(__dirname, "automatic", "webui.bat"), newtext)
    } else {
      // linux
      if (/amd/i.test(vendor)) {
        // lshqqytiger
        defaultArgs += "--precision full --no-half-vae --xformers --api"
      } else {
        defaultArgs += "--xformers --no-half-vae --api"
      }
      let text = await fs.promises.readFile(path.resolve(__dirname, "automatic", "webui-user.sh"), "utf8")
      let re = /^(#?)(export COMMANDLINE_ARGS=)(.+)$/m
      let newtext = text.replace(re, `$2"${defaultArgs}"`)
      await fs.promises.writeFile(path.resolve(__dirname, "automatic", "webui-user.sh"), newtext)
    }
  }
}
module.exports = Automatic