diff --git a/.github/workflows/test-unit.yml b/.github/workflows/test-unit.yml
index d0c8433..ab0a1a4 100644
--- a/.github/workflows/test-unit.yml
+++ b/.github/workflows/test-unit.yml
@@ -11,7 +11,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.10", "3.11"]
+        python-version: ["3.11"]
 
     steps:
     - name: Checkout code
@@ -50,7 +50,7 @@ jobs:
 
     - name: Upload coverage reports
       uses: codecov/codecov-action@v4
-      if: matrix.python-version == '3.10'
+      if: matrix.python-version == '3.11'
       with:
         file: ./coverage.xml
         flags: unittests
diff --git a/requirements.txt b/requirements.txt
index 9b1c790..6ee6ae9 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -22,6 +22,6 @@ pyyaml>=6.0
 tqdm>=4.65.0
 
 # For flash attention (optional for SA2VA performance)
-# torch>=2.0.0  # Usually provided by ComfyUI - uncomment if needed
+torch<=2.8.0
 torchvision
-# flash_attn  # Optional - improves SA2VA inference speed but not required
\ No newline at end of file
+# flash_attn  # Optional - improves SA2VA inference speed but not required