add model_file_name

huggingface · Mar 7, 2024 · 372ce59 · 372ce59
1 parent 29b9e85
commit 372ce59
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/pipelines.js b/src/pipelines.js
@@ -3019,6 +3019,7 @@ export async function pipeline(
         cache_dir = null,
         local_files_only = false,
         revision = 'main',
+        model_file_name = null
     } = {}
 ) {
     // Helper method to construct pipeline
@@ -3046,6 +3047,7 @@ export async function pipeline(
         cache_dir,
         local_files_only,
         revision,
+        model_file_name
     }
 
     const classes = new Map([