__init__.py 1.2 KB

1234567891011121314151617181920212223242526272829
  1. from aphrodite.transformers_utils.configs.chatglm import ChatGLMConfig
  2. from aphrodite.transformers_utils.configs.dbrx import DbrxConfig
  3. from aphrodite.transformers_utils.configs.eagle import EAGLEConfig
  4. # RWConfig is for the original tiiuae/falcon-40b(-instruct) and
  5. # tiiuae/falcon-7b(-instruct) models. Newer Falcon models will use the
  6. # `FalconConfig` class from the official HuggingFace transformers library.
  7. from aphrodite.transformers_utils.configs.falcon import RWConfig
  8. from aphrodite.transformers_utils.configs.granite import GraniteConfig
  9. from aphrodite.transformers_utils.configs.internvl import InternVLChatConfig
  10. from aphrodite.transformers_utils.configs.jais import JAISConfig
  11. from aphrodite.transformers_utils.configs.medusa import MedusaConfig
  12. from aphrodite.transformers_utils.configs.mlp_speculator import (
  13. MLPSpeculatorConfig)
  14. from aphrodite.transformers_utils.configs.mpt import MPTConfig
  15. from aphrodite.transformers_utils.configs.ultravox import UltravoxConfig
  16. __all__ = [
  17. "ChatGLMConfig",
  18. "DbrxConfig",
  19. "MPTConfig",
  20. "RWConfig",
  21. "InternVLChatConfig",
  22. "JAISConfig",
  23. "MLPSpeculatorConfig",
  24. "MedusaConfig",
  25. "UltravoxConfig",
  26. "EAGLEConfig",
  27. "GraniteConfig",
  28. ]