Merge branch 'preprocessor' into bricks

mybigday · Jun 17, 2024 · bc5b928 · bc5b928
2 parents 5112824 + 67917e0
commit bc5b928
Show file tree

Hide file tree

Showing 6 changed files with 343 additions and 61 deletions.
diff --git a/src/configs.js b/src/configs.js
@@ -78,6 +78,7 @@ export class PretrainedConfig {
         progress_callback = null,
         config = null,
         cache_dir = null,
+        subfolder = null,
         local_files_only = false,
         revision = 'main',
     } = {}) {
@@ -86,6 +87,7 @@ export class PretrainedConfig {
             progress_callback,
             config,
             cache_dir,
+            subfolder,
             local_files_only,
             revision,
         })

diff --git a/src/models.js b/src/models.js
@@ -743,6 +743,7 @@ export class PreTrainedModel extends Callable {
         progress_callback = null,
         config = null,
         cache_dir = null,
+        subfolder = null,
         local_files_only = false,
         revision = 'main',
         model_file_name = null,
@@ -754,6 +755,7 @@ export class PreTrainedModel extends Callable {
             progress_callback,
             config,
             cache_dir,
+            subfolder,
             local_files_only,
             revision,
             model_file_name,

diff --git a/src/pipelines.js b/src/pipelines.js
@@ -3182,6 +3182,7 @@ export async function pipeline(
         progress_callback = null,
         config = null,
         cache_dir = null,
+        subfolder = null,
         local_files_only = false,
         revision = 'main',
         session_options = {},
@@ -3211,6 +3212,7 @@ export async function pipeline(
         progress_callback,
         config,
         cache_dir,
+        subfolder,
         local_files_only,
         revision,
         session_options,