__init__.py 439 B

1234567891011121314
  1. from aphrodite.modeling.layers.fused_moe.fused_moe import (
  2. fused_experts, fused_moe, fused_topk, get_config_file_name, grouped_topk)
  3. from aphrodite.modeling.layers.fused_moe.layer import (FusedMoE,
  4. FusedMoEMethodBase)
  5. __all__ = [
  6. "fused_moe",
  7. "fused_topk",
  8. "fused_experts",
  9. "get_config_file_name",
  10. "grouped_topk",
  11. "FusedMoE",
  12. "FusedMoEMethodBase",
  13. ]