feat: add Exaone model support (#743)

* feat: add support for ExaoneForCausalLM models * progress bar
PygmalionAI · Sep 19, 2024 · 08711d2 · 08711d2
1 parent 81c28d2
commit 08711d2
Show file tree

Hide file tree

Showing 2 changed files with 620 additions and 0 deletions.
diff --git a/aphrodite/modeling/models/__init__.py b/aphrodite/modeling/models/__init__.py
@@ -77,6 +77,7 @@
     "ChameleonForConditionalGeneration": ("chameleon",
                                           "ChameleonForConditionalGeneration"),
     "SolarForCausalLM": ("solar", "SolarForCausalLM"),
+    "ExaoneForCausalLM": ("exaone", "ExaoneForCausalLM"),
 }
 
 _EMBEDDING_MODELS = {