minor type fix

IAHispano · Dec 3, 2024 · eff39c7 · eff39c7
1 parent 408e006
commit eff39c7
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/rvc/lib/algorithm/attentions.py b/rvc/lib/algorithm/attentions.py
@@ -201,10 +201,10 @@ class FFN(torch.nn.Module):
 
     def __init__(
         self,
-        in_channels=int,
-        out_channels=int,
-        filter_channels=int,
-        kernel_size=int,
+        in_channels: int,
+        out_channels: int,
+        filter_channels: int,
+        kernel_size: int,
         p_dropout: float = 0.0,
         activation: str = None,
         causal: bool = False,