浏览代码

Fix broken import statements in dataloader (#178)

Can Balioglu 1 年之前
父节点
当前提交
2ea56dd6dd
共有 1 个文件被更改,包括 2 次插入1 次删除
  1. 2 1
      src/seamless_communication/cli/m4t/finetune/dataloader.py

+ 2 - 1
src/seamless_communication/cli/m4t/finetune/dataloader.py

@@ -16,7 +16,8 @@ import torchaudio
 import torchaudio.compliance.kaldi as ta_kaldi
 from datasets import Dataset
 from datasets.distributed import split_dataset_by_node
-from fairseq2.models.nllb.tokenizer import NllbTokenizer, TextTokenEncoder
+from fairseq2.data.text import TextTokenEncoder
+from fairseq2.models.nllb import NllbTokenizer
 from torch import Tensor
 from torch.nn.functional import pad as pad_tensor
 from torch.utils.data import DataLoader