From 4a7ca0fc4d37ba0c6c1f87ed8b56858692d9cee7 Mon Sep 17 00:00:00 2001 From: Joshua Lochner Date: Fri, 31 May 2024 16:52:43 +0200 Subject: [PATCH] Support dtype, device, etc. in pipeline options --- src/pipelines.js | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/pipelines.js b/src/pipelines.js index dfbdc491b..7d8bd4875 100644 --- a/src/pipelines.js +++ b/src/pipelines.js @@ -3215,6 +3215,10 @@ export async function pipeline( cache_dir, local_files_only, revision, + device, + dtype, + model_file_name, + session_options, } const classes = new Map([