Skip to content

Commit

Permalink
Revise model registry for llava_hf and longva
Browse files Browse the repository at this point in the history
  • Loading branch information
kcz358 committed Jun 21, 2024
1 parent 7d6201f commit 120c474
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions lmms_eval/models/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,8 @@
"mplug_owl_video": "mplug_Owl",
"phi3v": "Phi3v",
"tinyllava": "TinyLlava",
"llava_hf": "LlavaHf",
"longva": "LongVA",
}

for model_name, model_class in AVAILABLE_MODELS.items():
Expand Down

0 comments on commit 120c474

Please sign in to comment.