fix: limiter threshold broadcast error (#5)

* fix: limiter threshold broadcast error * use dev version schema
DiffAPF · Aug 19, 2024 · 4f27963 · 4f27963
1 parent 4613c4c
commit 4f27963
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -1,7 +1,7 @@
 import setuptools
 
 NAME = "torchcomp"
-VERSION = "0.1.2"
+VERSION = "0.1.3dev"
 MAINTAINER = "Chin-Yun Yu"
 EMAIL = "[email protected]"
 

diff --git a/torchcomp/__init__.py b/torchcomp/__init__.py
@@ -197,5 +197,5 @@ def limiter_gain(
     zi = x.new_zeros(x.shape[0])
     lt = db2amp(threshold)
     x_peak = compressor_core(x.abs(), zi, rt, at)
-    f = F.relu(1 - lt / x_peak).neg() + 1
+    f = F.relu(1 - lt[:, None] / x_peak).neg() + 1
     return compressor_core(f, torch.ones_like(zi), at, rt)