wuff-mann commited on
Commit
a4af749
·
verified ·
1 Parent(s): b08171c

Upload 12 files

Browse files
Files changed (1) hide show
  1. mann_engram_en/router.py +2 -1
mann_engram_en/router.py CHANGED
@@ -96,7 +96,8 @@ class MANNEngramRouter:
96
  probs = F.softmax((t_scores - t_scores.max()) / temperature, dim=0).numpy()
97
  sorted_idx = np.argsort(probs)[::-1]
98
 
99
- selected_items, selected_probs, cum_p = [], 0.0
 
100
  for idx in sorted_idx:
101
  prob = float(probs[idx])
102
  cum_p += prob
 
96
  probs = F.softmax((t_scores - t_scores.max()) / temperature, dim=0).numpy()
97
  sorted_idx = np.argsort(probs)[::-1]
98
 
99
+ # 💡 就在这里:补上了缺失的那个空列表 []
100
+ selected_items, selected_probs, cum_p = [], [], 0.0
101
  for idx in sorted_idx:
102
  prob = float(probs[idx])
103
  cum_p += prob