From f24a943d937d8e53a3d3d2f1938377613ef0ccea Mon Sep 17 00:00:00 2001 From: kcz358 Date: Fri, 20 Dec 2024 11:28:01 +0000 Subject: [PATCH] Update --- lmms_eval/models/kino.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lmms_eval/models/kino.py b/lmms_eval/models/kino.py index be5d072f..b8c7ddc5 100644 --- a/lmms_eval/models/kino.py +++ b/lmms_eval/models/kino.py @@ -47,7 +47,7 @@ class Kino(lmms): def __init__( self, - pretrained: str = "kcz358/kino-7b-init", + pretrained: str = "lmms-lab/kino-7b-init", revision: str = "main", device: str = "cuda", dtype: Optional[Union[str, torch.dtype]] = "auto", @@ -89,7 +89,7 @@ def get_w(weights, keyword): eval_logger.info(f"Loaded audio_modal_projector weights from {pretrained_mlp_projector}. Incompatible keys: {incompatible_keys}") self.pretrained = pretrained - self._processor = KinoProcessor.from_pretrained(pretrained, revision=revision, trust_remote_code=trust_remote_code) + self._processor = KinoProcessor.from_pretrained("lmms-lab/kino-7b-init", revision=revision, trust_remote_code=trust_remote_code) # Pad from left for batched generation: https://huggingface.co/docs/transformers/v4.39.3/en/model_doc/llava#usage-tips self._processor.tokenizer.padding_side = "left" self._tokenizer = self._processor.tokenizer