This commit is contained in:
comfyanonymous
2025-09-12 18:57:04 -07:00
committed by GitHub
parent 2559dee492
commit 29bf807b0e
2 changed files with 2 additions and 2 deletions

View File

@@ -238,7 +238,7 @@ class Wav2Vec2Model(nn.Module):
device=device, dtype=dtype, operations=operations
)
def forward(self, x, sr=16000, mask_time_indices=None, return_dict=False):
def forward(self, x, mask_time_indices=None, return_dict=False):
x = torch.mean(x, dim=1)
if self.do_normalize: