rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-06-05 21:59:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
Fix transformers model changes
This commit is contained in:
Cohee
2024-07-08 22:16:10 +03:00
committed by GitHub
parent 00de522a64 968eb3790f
commit db4fe14011
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/transformers.mjs
View File

@@ -94,7 +94,11 @@ function getModelForTask(task) {
*/
async function getPipeline(task, forceModel = '') {
if (tasks[task].pipeline) {
return 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');
@@ -103,6 +107,7 @@ async function getPipeline(task, forceModel = '') {
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;
}