Эх сурвалжийг харах

Fix broken import statements in dataloader (#178)

Can Balioglu 1 жил өмнө
parent
commit
2ea56dd6dd

+ 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