complete arg docstring

Author: mchoi8739

doc/api/training/smp_versions/latest/smd_model_parallel_pytorch_tensor_parallel.rst

@@ -301,8 +301,7 @@ Tensor Parallelism Module APIs
          ``post_layernorm`` must be ``True``.
       -  ``post_layernorm``: If ``True``, inserts layer normalization at
          the output. At least one of ``pre_layernorm`` and
-         ``post_layernorm`` must be ``True``. (Available from 
-         the SageMaker model parallelism library v1.15.0.)
+         ``post_layernorm`` must be ``True``. 
       -  ``use_alibi`` (bool, default False): Activates Attention with 
          Linear Biases (ALiBi) for attention computation.
          ALiBi facilitates efficient extrapolation on input sequences 
@@ -314,6 +313,9 @@ Tensor Parallelism Module APIs
          details on the technique.
          (Available from 
          the SageMaker model parallelism library v1.15.0.)
+      -  ``alibi_bias_max`` (int, default 8): Defines the ALiBi base 
+      	 value for mask generation. (Available from 
+         the SageMaker model parallelism library v1.15.0.)
 
    -  **Methods:**