fix bug in rebase

ashors1 · ashors1 · commit 16e3abd9d595 · 2024-05-19T20:28:17.000-07:00
diff --git a/t5x/train.py b/t5x/train.py
@@ -280,7 +280,7 @@ def _verify_matching_vocabs(cfg: utils.DatasetConfig):
   )
 
   if prepare_train_iter_fn:
-    train_iter = utils.prepare_train_iter(
+    train_iter = prepare_train_iter_fn(
         train_iter,
         checkpoint_cfg=checkpoint_cfg,
         partitioner=partitioner,

Original file line number	Diff line number	Diff line change
`@@ -280,7 +280,7 @@ def _verify_matching_vocabs(cfg: utils.DatasetConfig):`
`280`	`280`	`)`
`281`	`281`
`282`	`282`	`if prepare_train_iter_fn:`
`283`		`- train_iter = utils.prepare_train_iter(`
	`283`	`+ train_iter = prepare_train_iter_fn(`
`284`	`284`	`train_iter,`
`285`	`285`	`checkpoint_cfg=checkpoint_cfg,`
`286`	`286`	`partitioner=partitioner,`