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:
Nawar Alsafar 2024-05-23 03:13:52 -07:00 committed by GitHub
parent db9250b3e8
commit 250fcd7ada
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions

View File

@ -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([