소스 검색

Update alignment_extractor.py (#439)

follow the same logic with layer id - 1 as the official unit extraction script
Ilia Kulikov 1 년 전
부모
커밋
66054d4278
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      src/seamless_communication/models/aligner/alignment_extractor.py

+ 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()