diff --git a/examples/data2vec/models/data2vec_audio.py b/examples/data2vec/models/data2vec_audio.py index 5cf5b53..ccaf4a4 100644 --- a/examples/data2vec/models/data2vec_audio.py +++ b/examples/data2vec/models/data2vec_audio.py @@ -423,7 +423,7 @@ class Data2VecAudioModel(BaseFairseqModel): # extracting both the source and the augmented audios from source source_audios = source[0] - aug_audios = source[1] + aug_audios = source[0] if self.feature_grad_mult > 0: features = self.feature_extractor(source_audios) #features diff --git a/setup.py b/setup.py index a7ce61a..3dc7b44 100644 --- a/setup.py +++ b/setup.py @@ -105,12 +105,12 @@ try: "fairseq/clib/libnat/edit_dist.cpp", ], ), - cpp_extension.CppExtension( - "alignment_train_cpu_binding", - sources=[ - "examples/operators/alignment_train_cpu.cpp", - ], - ), + # cpp_extension.CppExtension( + # "alignment_train_cpu_binding", + # sources=[ + # "examples/operators/alignment_train_cpu.cpp", + # ], + # ), ] ) if "CUDA_HOME" in os.environ:
