diff --git a/composer/algorithms/alibi/README.md b/composer/algorithms/alibi/README.md index 2f74860012..a72282b78c 100644 --- 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(