diff --git a/README.md b/README.md index f7f5775..a380996 100644 --- a/README.md +++ b/README.md @@ -27,7 +27,7 @@ source .venv/bin/activate ## Usage ```python import torch -from MoMoE import MoMoE +from momoe import MoMoE momoe = MoMoE( embedding_dim=2048, intermediate_dim=1024, @@ -73,4 +73,4 @@ We hope you enjoy our hyper-performant MoE! url={https://www.tilderesearch.com/blog/momoe}, note={Blog post} } -``` \ No newline at end of file +```