better comment

mosaicml · Sep 25, 2024 · 5004fe5 · 5004fe5
1 parent 8996de4
commit 5004fe5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/models/utils/test_tp_strategy.py b/tests/models/utils/test_tp_strategy.py
@@ -13,7 +13,8 @@
 
 
 def test_ffn_tp_strategy_layer_plan():
-    # Actual layer plan
+
+    # Actual layer plan from tp_strategy=fnn
     tp_config = {
         'strategy': 'ffn',
     }