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>
This commit is contained in:
parent
db9250b3e8
commit
250fcd7ada
|
@ -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([
|
||||
|
|
Loading…
Reference in New Issue