[Fix] use torch.cat instead of torch.concat to prevent entering the Autograd backends. (#4466)

2025-03-16 00:02:47 -07:00
parent 81f431eded
commit 1a3fa75f2f
14 changed files with 20 additions and 20 deletions
@@ -1289,7 +1289,7 @@ class MlpProjector(nn.Module):
            high_x, low_x = x_or_tuple
            high_x = self.high_up_proj(high_x)
            low_x = self.low_up_proj(low_x)
-            x = torch.concat([high_x, low_x], dim=-1)
+            x = torch.cat([high_x, low_x], dim=-1)
        else:
            x = x_or_tuple