Skip to content

Commit 93a8d47

Browse files
committed
Use built-in session release function
1 parent 6096b01 commit 93a8d47

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

src/models.js

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1100,9 +1100,7 @@ export class PreTrainedModel extends Callable {
11001100
async dispose() {
11011101
const promises = [];
11021102
for (const session of Object.values(this.sessions)) {
1103-
if (session?.handler?.dispose) {
1104-
promises.push(session.handler.dispose())
1105-
}
1103+
promises.push(session.release?.());
11061104
}
11071105
return await Promise.all(promises);
11081106
}

0 commit comments

Comments
 (0)