Spaces:

chenxie95
/

Language-Audio-Banquet

Sleeping

Jihuai commited on Sep 20

Commit

c69b591

1 Parent(s): a17f2e1

bug fix for querier

Files changed (1) hide show

core/models/e2e/querier/passt.py CHANGED Viewed

@@ -44,7 +44,7 @@ class Passt(nn.Module):
                 x = torch.mean(x, dim=1)
                 x = self.resample(x)
                 z = self.passt.get_audio_embedding_from_data(x, use_tensor=True)
-            if hasattr(x.query, "text"):
                 x = x.query.text
                 z = self.passt.get_text_embedding(x, use_tensor=True)

                 x = torch.mean(x, dim=1)
                 x = self.resample(x)
                 z = self.passt.get_audio_embedding_from_data(x, use_tensor=True)
+            elif hasattr(x.query, "text"):
                 x = x.query.text
                 z = self.passt.get_text_embedding(x, use_tensor=True)