razmars commited on
Commit
f872aa0
·
verified ·
1 Parent(s): 951555e

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. modeling_super_linear.py +0 -1
modeling_super_linear.py CHANGED
@@ -387,7 +387,6 @@ class SparseNoisyMoE(nn.Module):
387
  if hasattr(self, 'experts') and isinstance(getattr(self, 'experts', None), dict):
388
  # If experts are stored in a dictionary with meaningful keys
389
  expert_names = list(self.experts.keys())
390
- print(expert_names)
391
  plt.xticks(range(len(probs_np)), expert_names, rotation=45, ha='right')
392
  else:
393
  # Default numbering if expert names aren't available
 
387
  if hasattr(self, 'experts') and isinstance(getattr(self, 'experts', None), dict):
388
  # If experts are stored in a dictionary with meaningful keys
389
  expert_names = list(self.experts.keys())
 
390
  plt.xticks(range(len(probs_np)), expert_names, rotation=45, ha='right')
391
  else:
392
  # Default numbering if expert names aren't available