Spaces:

AIGC-Audio
/

Make-An-Audio-3

Running on Zero

guowenxiang commited on Jun 19, 2024

Commit

a310b79

verified ·

1 Parent(s): 4eb4d55

Update ldm/modules/diffusionmodules/flag_large_dit_moe.py

Files changed (1) hide show

ldm/modules/diffusionmodules/flag_large_dit_moe.py CHANGED Viewed

@@ -481,7 +481,7 @@ class FeedForward(nn.Module):
             dim, hidden_dim, bias=False
         )
-    @torch.compile
     def _forward_silu_gating(self, x1, x3):
         return F.silu(x1) * x3

             dim, hidden_dim, bias=False
         )
+    # @torch.compile
     def _forward_silu_gating(self, x1, x3):
         return F.silu(x1) * x3