Removed flash-attn as dependency because it's hard to install properly

2025-04-02 14:42:35 +02:00
parent 78c71ab8b4
commit e573b9ebe7
13 changed files with 147 additions and 118 deletions
--- a/setup.py
+++ b/setup.py
@@ -0,0 +1,29 @@
+from setuptools import setup, find_packages
+
+setup(
+    name="sia",
+    version="0.1.0",
+    packages=find_packages(),
+    entry_points={
+        'console_scripts': [
+            'sia=sia.__main__:main',
+        ],
+    },
+    install_requires=[
+        'torch>=2.0.0',
+        'accelerate>=0.26.0',
+        'aiohttp>=3.8.0',
+        'bitsandbytes>=0.41.0',
+        'dotenv-python>=0.0.1',
+        'huggingface_hub>=0.16.0',
+        'lxml>=4.9.0',
+        'mistral-common>=1.0.0',
+        'mistralai>=0.0.7',
+        'openai>=1.0.0',
+        'psutil>=5.9.0',
+        'python-dotenv>=1.0.0',
+        'tiktoken>=0.4.0',
+        'transformers>=4.30.0',
+    ],
+    python_requires='>=3.10',
+)