fix: hadamard tensors not included in wheel

aphrodite-engine · Feb 3, 2024 · f99eb2c · f99eb2c
1 parent 8b6790d
commit f99eb2c
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -411,8 +411,12 @@ def get_requirements() -> List[str]:
     install_requires=get_requirements(),
     ext_modules=ext_modules,
     cmdclass={"build_ext": BuildExtension},
-    package_data={"aphrodite-engine": ["aphrodite/endpoints/kobold/klite.embd",
-                                       "aphrodite/modeling/layers/quantization/hadamard.safetensors",
-                                       "py.typed"]},
+    package_data={
+        "aphrodite": [
+            "endpoints/kobold/klite.embd",
+            "modeling/layers/quantization/hadamard.safetensors",
+            "py.typed"
+        ]
+    },
     include_package_data=True,
 )