Updates after testing

2025-04-01 11:49:21 +00:00
parent 6f3d414d17
commit 130afbdd92
4 changed files with 28 additions and 11 deletions
--- a/setup.py
+++ b/setup.py
@@ -9,10 +9,11 @@ setup(
        'aiohttp>=3.8.0',
        'bitsandbytes>=0.41.0',
        'dotenv-python>=0.0.1',
+        'flash-attn>=2.0.0',
        'huggingface_hub>=0.16.0',
        'lxml>=4.9.0',
-        'mistralai>=0.0.7',
        'mistral-common>=1.0.0',
+        'mistralai>=0.0.7',
        'openai>=1.0.0',
        'psutil>=5.9.0',
        'python-dotenv>=1.0.0',