__init__.py 822 B

1234567891011121314151617181920
  1. from aphrodite.transformers_utils.configs.baichuan import BaiChuanConfig
  2. from aphrodite.transformers_utils.configs.chatglm import ChatGLMConfig
  3. from aphrodite.transformers_utils.configs.dbrx import DbrxConfig
  4. from aphrodite.transformers_utils.configs.mpt import MPTConfig
  5. from aphrodite.transformers_utils.configs.olmo import OLMoConfig
  6. from aphrodite.transformers_utils.configs.qwen import QWenConfig
  7. # RWConfig is for the original tiiuae/falcon-40b(-instruct) and
  8. # tiiuae/falcon-7b(-instruct) models. Newer Falcon models will use the
  9. # `FalconConfig` class from the official HuggingFace transformers library.
  10. from aphrodite.transformers_utils.configs.falcon import RWConfig
  11. __all__ = [
  12. "BaiChuanConfig",
  13. "ChatGLMConfig",
  14. "DbrxConfig",
  15. "MPTConfig",
  16. "OLMoConfig",
  17. "QWenConfig",
  18. "RWConfig",
  19. ]