[Bugfix] Use the correct repository revision for secondary artifact loaders (#57461)
Signed-off-by: Clinton Thomas <1033162+KernelClint@users.noreply.github.com> Co-authored-by: Lucas Bourtoule <35483370+dhalf@users.noreply.github.com> Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
This commit is contained in:
co-authored by
Lucas Bourtoule
mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
parent
d7f5722d7b
commit
cd63bf9996
@@ -529,7 +529,7 @@ class FunAudioChatProcessingInfo(BaseProcessingInfo):
|
||||
return TokenizersBackend.from_pretrained(
|
||||
self.model_id,
|
||||
subfolder="speech_tokenizer",
|
||||
revision=self.ctx.model_config.tokenizer_revision,
|
||||
revision=self.ctx.model_config.revision,
|
||||
)
|
||||
|
||||
def get_feature_extractor(self) -> WhisperFeatureExtractor:
|
||||
|
||||
@@ -277,8 +277,12 @@ def resolve_chat_template(
|
||||
if tools is None:
|
||||
chat_template = _try_get_processor_chat_template(
|
||||
tokenizer,
|
||||
revision=model_config.revision,
|
||||
code_revision=model_config.code_revision,
|
||||
revision=model_config.tokenizer_revision,
|
||||
code_revision=(
|
||||
model_config.code_revision
|
||||
if model_config.tokenizer == model_config.model
|
||||
else None
|
||||
),
|
||||
trust_remote_code=model_config.trust_remote_code,
|
||||
)
|
||||
if chat_template is not None:
|
||||
|
||||
Reference in New Issue
Block a user