diff --git a/plugins/accelerated-moe/src/fms_acceleration_moe/utils/scattermoe.py b/plugins/accelerated-moe/src/fms_acceleration_moe/utils/scattermoe.py index 152bee1d..d26139e5 100644 --- a/plugins/accelerated-moe/src/fms_acceleration_moe/utils/scattermoe.py +++ b/plugins/accelerated-moe/src/fms_acceleration_moe/utils/scattermoe.py @@ -35,7 +35,7 @@ except ImportError as e: raise ImportError( "kernel-hyperdrive PyPI package not found. Install it: " - "pip install -r plugins/accelerated-moe/requirements-scattermoe.txt" + "pip install -r plugins/accelerated-moe/requirements-khd.txt" ) from e # Local