__init__.py 954 B

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