updated requirements and test workflow config

.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

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
+# flash_attn  # Optional - improves SA2VA inference speed but not required