add flash attention

requirements.txt

@@ -1,7 +1,7 @@
 huggingface_hub
 einops
 sentence-transformers
-torch==2.2.0
+torch
 transformers
 openai
 python-dotenv
@@ -11,7 +11,6 @@ langchain-chroma
 unstructured[all-docs]
 libmagic
 gradio
-torch==2.2.0+cu121   # Ensure you're using the right PyTorch version with CUDA support
 flash-attn==2.6.3    # Flash attention module
 numpy<2              # Downgrade to avoid NumPy 2.0.1 conflicts
 pybind11>=2.12       # Ensure compatibility for modules needing pybind11