From eff39c789c4486579e1a342b4130d035d75fda99 Mon Sep 17 00:00:00 2001 From: Blaise Date: Tue, 3 Dec 2024 23:19:25 +0100 Subject: [PATCH] minor type fix --- rvc/lib/algorithm/attentions.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/rvc/lib/algorithm/attentions.py b/rvc/lib/algorithm/attentions.py index 542c6b02..d698afc8 100644 --- 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,