Add llava_hf back to registry

EvolvingLMMs-Lab · Jun 20, 2024 · 4587665 · 4587665
1 parent 3463651
commit 4587665
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/lmms_eval/models/__init__.py b/lmms_eval/models/__init__.py
@@ -29,6 +29,7 @@
     "mplug_owl_video": "mplug_Owl",
     "phi3v": "Phi3v",
     "tinyllava": "TinyLlava",
+    "llava_hf" : "LlavaHf",
 }
 
 for model_name, model_class in AVAILABLE_MODELS.items():