Skip to content

Commit 250fcd7

Browse files
NawarAxenova
andauthored
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 <[email protected]>
1 parent db9250b commit 250fcd7

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/pipelines.js

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3178,6 +3178,7 @@ export async function pipeline(
31783178
cache_dir = null,
31793179
local_files_only = false,
31803180
revision = 'main',
3181+
model_file_name = null,
31813182
} = {}
31823183
) {
31833184
// Helper method to construct pipeline
@@ -3205,6 +3206,7 @@ export async function pipeline(
32053206
cache_dir,
32063207
local_files_only,
32073208
revision,
3209+
model_file_name,
32083210
}
32093211

32103212
const classes = new Map([

0 commit comments

Comments
 (0)