adding back threads

cocktailpeanut · Mar 22, 2023 · a203853 · a203853
1 parent 6b1c93e
commit a203853
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/bin/web/index.js b/bin/web/index.js
@@ -15,7 +15,9 @@ const start = (port, home) => {
  app.set('view engine', 'ejs');
  app.set('views', path.resolve(__dirname, "views"))
  app.get("/", (req, res) => {
- res.render("index")
+ res.render("index", {
+ threads: os.cpus().length
+ })
  })
 
  const defaultDir = path.join(__dirname, "prompts");
@@ -69,4 +71,4 @@ const start = (port, home) => {
  console.log(`Server running on http:https://localhost:${port}/`)
  })
 }
-module.exports = start
+module.exports = start
diff --git a/bin/web/views/index.ejs b/bin/web/views/index.ejs
@@ -229,7 +229,7 @@
  const fields = [{
  key: "debug",
  type: "checkbox"
- }, "n_predict", "repeat_last_n", "repeat_penalty", "top_k", "top_p", "temp", "seed"].map((key) => {
+ }, "n_predict", "repeat_last_n", "repeat_penalty", "top_k", "top_p", "temp", "seed", "threads"].map((key) => {
  if (typeof key === "string") {
  return `<div class='kv'>
 <label>${key}</label>