Fix typo in embed.py docs (#677)

TransformerLensOrg · Jul 15, 2024 · b1c9381 · b1c9381
1 parent cec7ed3
commit b1c9381
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/transformer_lens/components/embed.py b/transformer_lens/components/embed.py
@@ -1,6 +1,6 @@
 """Hooked Transformer Embed Component.
 
-This module contains all the component :class:`BertMLMHead`.
+This module contains all the component :class:`Embed`.
 """
 from typing import Dict, Union