Fix hardcoded CUDA call in StyleCLIP fused_act for curly hair editing

editings/styleclip/models/stylegan2/op/fused_act.py

@@ -22,11 +22,13 @@ class FusedLeakyReLU(nn.Module):
 
 def fused_leaky_relu(input, bias, negative_slope=0.2, scale=2**0.5):
     rest_dim = [1] * (input.ndim - bias.ndim - 1)
-    input = input.cuda()
+    # Use the same device as input instead of hardcoded CUDA
+    device = input.device
+    input = input.to(device)
     if input.ndim == 3:
         return (
             F.leaky_relu(
-                input + bias.view(1, *rest_dim, bias.shape[0]),
+                input + bias.view(1, *rest_dim, bias.shape[0]).to(device),
                 negative_slope=negative_slope,
             )
             * scale
@@ -34,7 +36,7 @@ def fused_leaky_relu(input, bias, negative_slope=0.2, scale=2**0.5):
     else:
         return (
             F.leaky_relu(
-                input + bias.view(1, bias.shape[0], *rest_dim),
+                input + bias.view(1, bias.shape[0], *rest_dim).to(device),
                 negative_slope=negative_slope,
             )
             * scale