Update

EvolvingLMMs-Lab · Dec 22, 2024 · f24a943 · f24a943
1 parent 4a5cdfe
commit f24a943
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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