fix typo in example (#1259)

mosaicml · Jul 16, 2022 · b92c8eb · b92c8eb
1 parent 1be8801
commit b92c8eb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/composer/algorithms/alibi/README.md b/composer/algorithms/alibi/README.md
@@ -66,7 +66,7 @@ alibi = Alibi(
     attention_module_name="transformers.models.gpt2.modeling_gpt2.GPT2Attention",
     attr_to_replace="_attn",
     alibi_attention="composer.algorithms.alibi._gpt2_alibi._attn",
-    mask_replacement_function="composer.algorithms.alibi.gpt2_alibi.enlarge_mask"
+    mask_replacement_function="composer.algorithms.alibi._gpt2_alibi.enlarge_mask"
 )
 
 trainer = Trainer(