Add model file name (#594)

* add model_file_name * Allow developers to configure model_file_name (only works for decoder-only models for now) * remove prettier formatting * merge with 2.15.1 * add model_file_name * Apply suggestions from code review --------- Co-authored-by: Joshua Lochner <admin@xenova.com>
xenova · May 23, 2024 · 250fcd7 · 250fcd7
1 parent db9250b
commit 250fcd7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/pipelines.js b/src/pipelines.js
@@ -3178,6 +3178,7 @@ export async function pipeline(
         cache_dir = null,
         local_files_only = false,
         revision = 'main',
+        model_file_name = null,
     } = {}
 ) {
     // Helper method to construct pipeline
@@ -3205,6 +3206,7 @@ export async function pipeline(
         cache_dir,
         local_files_only,
         revision,
+        model_file_name,
     }
 
     const classes = new Map([