Update models.py

Bebra777228 · web-flow · commit 4650870707e5 · 2024-08-05T00:01:16.000+05:00
diff --git a/src/infer_pack/models.py b/src/infer_pack/models.py
@@ -17,7 +17,7 @@
 class TextEncoder(nn.Module):
     def __init__(
         self,
-        input_dim
+        input_dim,
         out_channels,
         hidden_channels,
         filter_channels,
