Spaces:
Sleeping
Sleeping
Upload 12 files
Browse files- 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 |
-
|
|
|
|
| 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
|