fix pypi dependences

LAION-AI · Apr 19, 2023 · 740d350 · 740d350
1 parent 386e9f8
commit 740d350
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -3,7 +3,7 @@ requires = ["setuptools>=61.0"]
 build-backend = "setuptools.build_meta"
 [project]
 name = "laion_clap"
-version = "1.1.3"
+version = "1.1.4"
 authors = [
   { name="Ke Chen", email="[email protected]" },
   { name="Yusong Wu" },

diff --git a/src/laion_clap/__init__.py b/src/laion_clap/__init__.py
@@ -2,4 +2,4 @@
 import sys
 dir_path = os.path.dirname(os.path.abspath(__file__))
 sys.path.append(dir_path)
-from hook import CLAP_Module
+from .hook import CLAP_Module
diff --git a/src/laion_clap/unit_test.py b/src/laion_clap/unit_test.py
@@ -20,7 +20,7 @@ def float32_to_int16(x):
     x = np.clip(x, a_min=-1., a_max=1.)
     return (x * 32767.).astype(np.int16)
 
-model = laion_clap.CLAP_Module(enable_fusion=True)
+model = laion_clap.CLAP_Module(enable_fusion=False)
 model.load_ckpt()
 
 # Directly get audio embeddings from audio files