Merge pull request #2495 from Risenafis/fix-change-model Fix transformers model changes

db4fe14011ef508d4090ccd52c2a7902c5a55fd7

Cohee <18619528+Cohee1207@users.noreply.github.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;
+}