fix gmlp comment

Author: vpj

docs/transformers/gmlp/index.html

@@ -102,10 +102,12 @@ <h2>gMLP Block</h2>
             <div class='section-link'>
                 <a href='#section-2'>#</a>
             </div>
-            <p> <code class="highlight"><span></span><span class="n">d_model</span></code>
- is the dimensionality (<span class="katex"><span class="katex-html" aria-hidden="true"><span class="base"><span class="strut" style="height:0.69444em;vertical-align:0em;"></span><span class="mord mathnormal">d</span></span></span></span>) of <span class="katex"><span class="katex-html" aria-hidden="true"><span class="base"><span class="strut" style="height:0.68333em;vertical-align:0em;"></span><span class="mord mathnormal" style="margin-right:0.07847em;">X</span></span></span></span> <code class="highlight"><span></span><span class="n">d_ffn</span></code>
- is the dimensionality of <span class="katex"><span class="katex-html" aria-hidden="true"><span class="base"><span class="strut" style="height:0.68333em;vertical-align:0em;"></span><span class="mord mathnormal" style="margin-right:0.07153em;">Z</span></span></span></span> <code class="highlight"><span></span><span class="n">seq_len</span></code>
- is the length of the token sequence (<span class="katex"><span class="katex-html" aria-hidden="true"><span class="base"><span class="strut" style="height:0.43056em;vertical-align:0em;"></span><span class="mord mathnormal">n</span></span></span></span>)</p>
+            <ul><li><code class="highlight"><span></span><span class="n">d_model</span></code>
+ is the dimensionality (<span class="katex"><span class="katex-html" aria-hidden="true"><span class="base"><span class="strut" style="height:0.69444em;vertical-align:0em;"></span><span class="mord mathnormal">d</span></span></span></span>) of <span class="katex"><span class="katex-html" aria-hidden="true"><span class="base"><span class="strut" style="height:0.68333em;vertical-align:0em;"></span><span class="mord mathnormal" style="margin-right:0.07847em;">X</span></span></span></span> </li>
+<li><code class="highlight"><span></span><span class="n">d_ffn</span></code>
+ is the dimensionality of <span class="katex"><span class="katex-html" aria-hidden="true"><span class="base"><span class="strut" style="height:0.68333em;vertical-align:0em;"></span><span class="mord mathnormal" style="margin-right:0.07153em;">Z</span></span></span></span> </li>
+<li><code class="highlight"><span></span><span class="n">seq_len</span></code>
+ is the length of the token sequence (<span class="katex"><span class="katex-html" aria-hidden="true"><span class="base"><span class="strut" style="height:0.43056em;vertical-align:0em;"></span><span class="mord mathnormal">n</span></span></span></span>)</li></ul>
 
         </div>
         <div class='code'>

labml_nn/transformers/gmlp/__init__.py

@@ -47,9 +47,9 @@ class GMLPBlock(nn.Module):
 
     def __init__(self, d_model: int, d_ffn: int, seq_len: int):
         """
-        `d_model` is the dimensionality ($d$) of $X$
-        `d_ffn` is the dimensionality of $Z$
-        `seq_len` is the length of the token sequence ($n$)
+        * `d_model` is the dimensionality ($d$) of $X$
+        * `d_ffn` is the dimensionality of $Z$
+        * `seq_len` is the length of the token sequence ($n$)
         """
         super().__init__()
         # Normalization layer fro Pre-Norm