Add option to use external data format in pipeline initialization

huggingface · Feb 1, 2025 · 625ecc7 · 625ecc7
1 parent 6f43f24
commit 625ecc7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/pipelines.js b/src/pipelines.js
@@ -3301,6 +3301,7 @@ export async function pipeline(
         dtype = null,
         model_file_name = null,
         session_options = {},
+        use_external_data_format = false,
     } = {}
 ) {
     // Helper method to construct pipeline
@@ -3331,6 +3332,7 @@ export async function pipeline(
         dtype,
         model_file_name,
         session_options,
+        use_external_data_format,
     }
 
     const classes = new Map([