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

Update alignment_extractor.py (#439)

follow the same logic with layer id - 1 as the official unit extraction script
Ilia Kulikov 1 жил өмнө
parent
commit
66054d4278

+ 1 - 1
src/seamless_communication/models/aligner/alignment_extractor.py

@@ -91,7 +91,7 @@ class AlignmentExtractor(nn.Module):
         assert isinstance(
             self.unit_extractor, UnitExtractor
         ), "Unit extractor is required to get units from audio tensor"
-        units = self.unit_extractor.predict(audio, self.unit_extractor_output_layer)
+        units = self.unit_extractor.predict(audio, self.unit_extractor_output_layer - 1)
         return units
 
     @torch.inference_mode()