Add trust_remote_code parameter to tokenizer (#3631)
This commit is contained in:
parent
13b3e7e6a8
commit
f91ea1e9a6
1 changed files with 3 additions and 0 deletions
|
|
@ -780,6 +780,7 @@ class FastBaseModel:
|
|||
token = token,
|
||||
language = whisper_language,
|
||||
task = whisper_task,
|
||||
trust_remote_code = trust_remote_code,
|
||||
)
|
||||
else:
|
||||
try:
|
||||
|
|
@ -787,12 +788,14 @@ class FastBaseModel:
|
|||
tokenizer_name,
|
||||
padding_side = "left",
|
||||
token = token,
|
||||
trust_remote_code = trust_remote_code,
|
||||
)
|
||||
except:
|
||||
tokenizer = get_auto_processor(
|
||||
tokenizer_name,
|
||||
padding_side = "left",
|
||||
token = token,
|
||||
trust_remote_code = trust_remote_code,
|
||||
)
|
||||
if hasattr(tokenizer, "tokenizer"):
|
||||
__tokenizer = tokenizer.tokenizer
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue