Update modeling_super_linear.py

modeling_super_linear.py

@@ -386,6 +386,7 @@ class SparseNoisyMoE(nn.Module):
         if hasattr(self, 'experts') and isinstance(getattr(self, 'experts', None), dict):
             # If experts are stored in a dictionary with meaningful keys
             expert_names = list(self.experts.keys())
+            print(expert_names)
             plt.xticks(range(len(probs_np)), expert_names, rotation=45, ha='right')
         else:
             # Default numbering if expert names aren't available