Skip to content

Commit 5143678

Browse files
committed
fix(ml): debug info
1 parent 215a155 commit 5143678

File tree

1 file changed

+30
-26
lines changed
  • src/services/backend/workers/background

1 file changed

+30
-26
lines changed

src/services/backend/workers/background/worker.ts

Lines changed: 30 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -94,41 +94,45 @@ const createBackgroundWorkerApi = () => {
9494

9595
return pipeline(model.name, model.model, {
9696
progress_callback: (progressData: any) => {
97-
const {
98-
status,
99-
progress,
100-
// name: modelName,
101-
loaded,
102-
total,
103-
} = progressData;
104-
105-
const message = loaded
106-
? `${model.model} - ${loaded}/${total} bytes`
107-
: model.model;
108-
const progressItem = {
109-
status,
110-
message,
111-
done: ['done', 'ready', 'error'].some((s) => s === status),
112-
};
113-
console.log('progress_callback', name, progressData);
114-
115-
if (name === 'featureExtractor' && status === 'done') {
116-
getEmbeddingInstance$.next(getEmbedding);
97+
try {
98+
const {
99+
status,
100+
progress,
101+
// name: modelName,
102+
loaded,
103+
total,
104+
} = progressData;
105+
106+
const message = loaded
107+
? `${model.model} - ${loaded}/${total} bytes`
108+
: model.model;
109+
const progressItem = {
110+
status,
111+
message,
112+
done: ['done', 'ready', 'error'].some((s) => s === status),
113+
};
114+
console.log('progress_callback', name, progressData);
115+
116+
if (progress) {
117+
progressItem.progress = Math.round(progress);
118+
}
119+
120+
broadcastApi.postMlSyncEntryProgress(name, progressItem);
121+
} catch (e) {
122+
console.log('-------progresss error', name, e.toString());
117123
}
118-
119-
if (progress) {
120-
progressItem.progress = Math.round(progress);
121-
}
122-
123-
broadcastApi.postMlSyncEntryProgress(name, progressItem);
124124
},
125125
}).then((model) => {
126126
console.log('----model', name, typeof model);
127+
getEmbeddingInstance$.next(getEmbedding);
128+
127129
mlInstances[name] = model;
128130

129131
return model;
130132
});
131133
}
134+
console.log(`${name} - already loaded`);
135+
getEmbeddingInstance$.next(getEmbedding);
132136

133137
return mlInstances[name];
134138
};

0 commit comments

Comments
 (0)