no numba for now

kyegomez · Jul 17, 2023 · 9ebf75a · 9ebf75a
1 parent 11ef5c1
commit 9ebf75a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/pegasus/main.py b/pegasus/main.py
@@ -1,14 +1,14 @@
 from concurrent.futures import ProcessPoolExecutor, as_completed
-from numba import njit
-from joblib import Memory
+# from numba import njit
+# from joblib import Memory
 import numpy as np
 
 from pegasus.embedding_functions import MultiModalEmbeddingFunction
 
-memory = Memory("PegasusStore", verbose=0)
+# memory = Memory("PegasusStore", verbose=0)
 
-@memory.cache
-@njit
+# @memory.cache
+# # @njit
 def optimized_embedding_function(modality, data):
     return MultiModalEmbeddingFunction(modality)(data)
 

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'pegasusX',
   packages = find_packages(exclude=[]),
-  version = '0.2.9',
+  version = '0.3.0',
   license='MIT',
   description = 'pegasus - Pytorch',
   author = 'Kye Gomez',