fix transformers model changes

968eb3790f32ae444740d240cf2e5a12f7fab5c8

Risenafis <reverse.hierarchy@gmail.com>

Signed

src/transformers.mjs+5 -0 Viewed

  */
 async function getPipeline(task, forceModel = '') {
     if (tasks[task].pipeline) {
+        if (forceModel === '' || tasks[task].currentModel === forceModel) {
             return tasks[task].pipeline;
+        }
+        console.log('Disposing transformers.js pipeline for for task', task, 'with model', tasks[task].currentModel);
+        await tasks[task].pipeline.dispose();
+    }
     const cache_dir = path.join(process.cwd(), 'cache');
     const model = forceModel || getModelForTask(task);

     console.log('Initializing transformers.js pipeline for task', task, 'with model', model);
     const instance = await pipeline(task, model, { cache_dir, quantized: tasks[task].quantized ?? true, local_files_only: localOnly });
     tasks[task].pipeline = instance;
+    tasks[task].currentModel = model;
     return instance;
+}