Try dropdown again

Author: mikaylagawarecki

intermediate_source/transformer_building_blocks.py

@@ -253,15 +253,16 @@ def forward(self,
 
         return attn_output
 
+# .. dropdown::
+
+    ###############################################################################
+    # Utilities
+    # =========
+    # In this section, we include a utility to generate semi-realistic data using
+    # Zipf distribution for sentence lengths. This is used to generate the nested
+    # query, key and value tensors. We also include a benchmark utility.
 
-###############################################################################
-# Utilities
-# =========
-# In this section, we include a utility to generate semi-realistic data using
-# Zipf distribution for sentence lengths. This is used to generate the nested
-# query, key and value tensors. We also include a benchmark utility.
 
-# .. dropdown::
     import numpy as np
 
     def zipf_sentence_lengths(alpha: float, batch_size: int) -> torch.Tensor: