Support dtype, device, etc. in pipeline options

huggingface · May 31, 2024 · 4a7ca0f · 4a7ca0f
1 parent 9c48305
commit 4a7ca0f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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([