Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/JinjieNi/MegaDLMs
/ endpoints
Endpoints
8 in github.com/JinjieNi/MegaDLMs
⨍
Functions
4,219
◇
Types & classes
650
↳
Endpoints
8
Route
collect_train_test_metrics
click.option("--output-path", required=False, type=str, help="Path to write golden values")
tests/functional_tests/python_test_utils/get_test_results_from_tensorboard_logs.py:None
Route
main
click.option( "--output-path", required=True, type=str, help="Directory where the functional t
tests/test_utils/python_scripts/generate_local_jobs.py:None
Route
main
click.option("--output-path", required=True, type=str, help="Path to write GitLab job to")
tests/test_utils/python_scripts/generate_jet_trigger_job.py:None
Route
test_completions
unittest.mock.patch('megatron.inference.endpoints.completions.get_tokenizer')
tests/unit_tests/test_inference.py:None
Route
test_legacy_moe_layer
pytest.mark.parametrize("moe_token_dispatcher_type", ["allgather", "alltoall"])
tests/unit_tests/transformer/moe/test_moe_layer.py:None
Route
test_moe_with_late_initialize
pytest.mark.parametrize("moe_token_dispatcher_type", ["allgather", "alltoall"])
tests/unit_tests/transformer/moe/test_moe_layer.py:None
Route
test_router_forward
pytest.mark.parametrize("moe_router_pre_softmax", [(True), (False)])
tests/unit_tests/transformer/moe/test_routers.py:None
Route
test_te_moe_layer
pytest.mark.parametrize("moe_token_dispatcher_type", ["allgather", "alltoall"])
tests/unit_tests/transformer/moe/test_moe_layer.py:None