trying a quick fix

networks/op/fused_act.py

@@ -19,7 +19,6 @@ else:
         ],
     )
 
-
 class FusedLeakyReLUFunctionBackward(Function):
     @staticmethod
     def forward(ctx, grad_output, out, bias, negative_slope, scale):
@@ -128,3 +127,5 @@ def fused_leaky_relu(input, bias=None, negative_slope=0.2, scale=2 ** 0.5):
         return FusedLeakyReLUFunction.apply(
             input.contiguous(), bias, negative_slope, scale
         )
+
+torch.compiler.allow_in_graph(fused.fused_bias_act)

networks/op/upfirdn2d.py

@@ -210,3 +210,5 @@ def upfirdn2d_native(
     out_w = (in_w * up_x + pad_x0 + pad_x1 - kernel_w + down_x) // down_x
 
     return out.view(-1, channel, out_h, out_w)
+
+torch.compiler.allow_in_graph(upfirdn2d_op.upfirdn2d)