tokenizer_name

This commit is contained in:
Daniel Han 2024-11-21 02:28:00 -08:00
commit 225c617f48
2 changed files with 2 additions and 0 deletions

View file

@ -493,6 +493,7 @@ class FastVisionModel(FastBaseVisionModel):
trust_remote_code = trust_remote_code,
revision = revision if not is_peft else None,
model_types = model_types,
tokenizer_name = tokenizer_name,
*args, **kwargs,
)

View file

@ -49,6 +49,7 @@ class FastBaseVisionModel:
device_map = "sequential",
trust_remote_code = False,
model_types = None,
tokenizer_name = None,
**kwargs,
):
if trust_remote_code: