Skip to content

Commit 2934dfe

Browse files
lingvo-botcopybara-github
authored andcommitted
Fix decoding.
PiperOrigin-RevId: 480135695
1 parent 48e4f56 commit 2934dfe

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

lingvo/core/gshard_layers.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2638,7 +2638,6 @@ def _MaybeSplit(x):
26382638
expert_capacity_dim, capacity_factor, group_size_dim, experts_dim,
26392639
tf.get_default_graph().get_name_scope())
26402640
tpu_summary.scalar('expert_capacity', expert_capacity_dim)
2641-
26422641
# GEC
26432642
_, token_indices = tf.math.top_k(
26442643
token_scores, k=expert_capacity_dim, sorted=False)

0 commit comments

Comments
 (0)