Skip to content

Commit 4996ddb

Browse files
committed
Update
1 parent 0effbb3 commit 4996ddb

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

lmms_eval/models/kino.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ class Kino(lmms):
4747

4848
def __init__(
4949
self,
50-
pretrained: str = "kcz358/kino-7b-init",
50+
pretrained: str = "Evo-LMM/kino-7b-init",
5151
revision: str = "main",
5252
device: str = "cuda",
5353
dtype: Optional[Union[str, torch.dtype]] = "auto",
@@ -89,7 +89,7 @@ def get_w(weights, keyword):
8989
eval_logger.info(f"Loaded audio_modal_projector weights from {pretrained_mlp_projector}. Incompatible keys: {incompatible_keys}")
9090

9191
self.pretrained = pretrained
92-
self._processor = KinoProcessor.from_pretrained(pretrained, revision=revision, trust_remote_code=trust_remote_code)
92+
self._processor = KinoProcessor.from_pretrained("Evo-LMM/kino-7b-init", revision=revision, trust_remote_code=trust_remote_code)
9393
# Pad from left for batched generation: https://huggingface.co/docs/transformers/v4.39.3/en/model_doc/llava#usage-tips
9494
self._processor.tokenizer.padding_side = "left"
9595
self._tokenizer = self._processor.tokenizer

0 commit comments

Comments
 (0)