From 1afbc5a80847b54ef29ca42bbd4ebb6060da0e19 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <2225664821@qq.com>
Date: Sun, 20 Aug 2023 15:12:06 +0800
Subject: [PATCH] Update attentions.py

---
 attentions.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/attentions.py b/attentions.py
index 634897f..4a42a1a 100644
--- a/attentions.py
+++ b/attentions.py
@@ -44,10 +44,10 @@ class Encoder(nn.Module):
     self.p_dropout = p_dropout
     self.window_size = window_size
     if isflow:
-      cond_layer = torch.nn.Conv1d(kwargs["gin_channels"], 2*hidden_channels*n_layers, 1)
+      cond_layer = torch.nn.Conv1d(256, 2*hidden_channels*n_layers, 1)
       self.cond_pre = torch.nn.Conv1d(hidden_channels, 2*hidden_channels, 1)
       self.cond_layer = weight_norm(cond_layer, name='weight')
-      self.gin_channels = kwargs["gin_channels"]
+      self.gin_channels = 256
     self.drop = nn.Dropout(p_dropout)
     self.attn_layers = nn.ModuleList()
     self.norm_layers_1 = nn.ModuleList()