diff --git a/.DS_Store b/.DS_Store
deleted file mode 100644
index 5008ddfc..00000000
Binary files a/.DS_Store and /dev/null differ
diff --git a/README.md b/README.md
index 2b1efcb0..e66c4bf8 100644
--- a/README.md
+++ b/README.md
@@ -66,7 +66,8 @@ We provide several variants for each of the components in the unlearning pipelin
 ```bash
 conda create -n unlearning python=3.11
 conda activate unlearning
-pip install .[flash-attn]
+pip install .
+pip install --no-build-isolation flash-attn==2.6.3
 ```
 
 ### 💾 Data Setup
@@ -176,4 +177,4 @@ If you use OpenUnlearning in your research, please cite:
 ---
 
 ## 📄 License
-This project is licensed under the MIT License. See the [`LICENSE`](LICENSE) file for details.
\ No newline at end of file
+This project is licensed under the MIT License. See the [`LICENSE`](LICENSE) file for details.
diff --git a/requirements.txt b/requirements.txt
index 01abf456..147f5155 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,6 @@
 huggingface-hub==0.29.1
 transformers==4.45.1
+numpy==2.2.3
 hydra-core==1.3
 hydra_colorlog==1.2.0
 torch==2.4.1
@@ -11,4 +12,4 @@ pre-commit==4.0.1
 scipy==1.14.1
 tensorboard==2.18.0
 scikit-learn==1.5.2
-deepspeed==0.15.4
\ No newline at end of file
+deepspeed==0.15.4
diff --git a/setup.py b/setup.py
index 4dcdef9a..79c6dbc9 100644
--- a/setup.py
+++ b/setup.py
@@ -16,7 +16,6 @@
     packages=find_packages(),
     install_requires=requirements,  # Uses requirements.txt
     extras_require={
-        "flash-attn": ["flash-attn==2.6.3"],  # Optional installation
         "dev": ["pre-commit==4.0.1"],  # Install using `pip install .[dev]`
     },
     python_requires=">=3.11",