@@ -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