Rename exllama extensions (#4)

casper-hansen · Jan 21, 2024 · 4bb0c02 · 4bb0c02
1 parent e6398f8
commit 4bb0c02
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -158,7 +158,7 @@ def get_compute_capabilities():
 
 extensions.append(
     CUDAExtension(
-        "exllama_kernels",
+        "exl_ext",
         [
             "awq_ext/exllama/exllama_ext.cpp",
             "awq_ext/exllama/cuda_buffers.cu",
@@ -172,7 +172,7 @@ def get_compute_capabilities():
 )
 extensions.append(
     CUDAExtension(
-        "exllamav2_kernels",
+        "exlv2_ext",
         [
             "awq_ext/exllamav2/ext.cpp",
             "awq_ext/exllamav2/cuda/q_matrix.cu",