Report transformers.js env.

package.json

@@ -1,6 +1,6 @@
 {
   "name": "localm",
-  "version": "1.1.6",
+  "version": "1.1.7",
   "description": "",
   "main": "chat-full.js",
   "scripts": {

src/app/boot-app.js

@@ -19,8 +19,9 @@ export var worker;
 export async function bootApp() {
   const { chatLog, chatInput } = initHTML();
   worker = workerConnection();
-  worker.loaded.then(async () => {
-    outputMessage('Models...');
+  worker.loaded.then(async ({ env }) => {
+    outputMessage(
+      'transformers.js\n\n```JSON\n' + JSON.stringify(env, null, 2) + '\n```');
     const models = await worker.listModels();
     outputMessage('Available models: ' + models.join(', '));
   });

src/app/worker-connection.js

@@ -6,7 +6,7 @@ export function workerConnection() {
   const workerLoaded = loadWorker();
 
   const connection = {
-    loaded: workerLoaded.then(() => { }),
+    loaded: workerLoaded.then(worker => ({ env: worker.env })),
     listModels,
     loadModel,
     runPrompt
@@ -33,7 +33,7 @@ export function workerConnection() {
           const msg = ev.data || {};
           if (msg && msg.type === 'ready') {
             ready = true;
-            resolve({ worker, pending, send });
+            resolve({ worker, pending, send, env: msg.env });
             return;
           }
 
@@ -71,15 +71,15 @@ export function workerConnection() {
 
   async function listModels() {
     await workerLoaded;
-  const { send } = await workerLoaded;
-  return send({ type: 'listModels' });
+    const { send } = await workerLoaded;
+    return send({ type: 'listModels' });
   }
 
   /** @param {string} modelName */
   async function loadModel(modelName) {
     await workerLoaded;
-  const { send } = await workerLoaded;
-  return send({ type: 'loadModel', modelName });
+    const { send } = await workerLoaded;
+    return send({ type: 'loadModel', modelName });
   }
 
   /**

src/worker/boot-worker.js

@@ -15,7 +15,7 @@ export function bootWorker() {
 
 
   // signal ready to main thread (worker script loaded; model runtime may still be pending)
-  self.postMessage({ type: 'ready' });
+  self.postMessage({ type: 'ready', env: modelCache.env, backend: modelCache.backend });
 
   // handle incoming requests from the UI thread
   self.addEventListener('message', handleMessage);

src/worker/model-cache.js

@@ -1,5 +1,7 @@
 // @ts-check
 
+import { pipeline, env } from '@huggingface/transformers';
+
 import { loadModelCore } from './load-model-core';
 
 export class ModelCache {
@@ -7,9 +9,11 @@ export class ModelCache {
   /** @type {import('@huggingface/transformers').DeviceType | undefined} */
   backend = undefined;
 
+  env = env;
+
   knownModels = [
-    'Xenova/phi-3-mini-4k-instruct',
     'Xenova/llama2.c-stories15M', // nonsense
+    'Xenova/phi-3-mini-4k-instruct', // huge
     'Xenova/all-MiniLM-L6-v2', // unsupported model type: bert
     'Xenova/phi-1.5', // gated
     'Qwen/Qwen2.5-3B', // cannot be loaded