Browse Source

Fix broken import statements in dataloader (#178)

Can Balioglu 1 năm trước cách đây
mục cha
commit
2ea56dd6dd
1 tập tin đã thay đổi với 2 bổ sung1 xóa
  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