Add sae hooked llava

EvolvingLMMs-Lab · Sep 16, 2024 · 0a6889e · 0a6889e
1 parent a2a881c
commit 0a6889e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/lmms_eval/models/__init__.py b/lmms_eval/models/__init__.py
@@ -25,6 +25,7 @@
     "llava": "Llava",
     "llava_hf": "LlavaHf",
     "llava_onevision": "Llava_OneVision",
+    "llava_sae_hooked": "LlavaSaeHooked",
     "llava_sglang": "LlavaSglang",
     "llava_vid": "LlavaVid",
     "longva": "LongVA",