Skip to content

Commit 175d61a

Browse files
committed
Fix PyTorch version conflict in requirements.transformers.txt
Updated requirements.transformers.txt to match requirements.sam.txt: - torch: >=2.0.1,<2.7.0 → >=2.8.0 - torchvision: >=0.15.0 → >=0.23.0 - Added: flash-attn>=2.8.2 This resolves the dependency conflict that was causing builds to fail.
1 parent 6e7b94d commit 175d61a

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

requirements/requirements.transformers.txt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
1-
# TODO: update to 2.8.0 once pre-built flashattn is available
2-
torch>=2.0.1,<2.7.0
3-
torchvision>=0.15.0
1+
torch>=2.8.0
2+
torchvision>=0.23.0
3+
flash-attn>=2.8.2
44
transformers>=4.53.3,<4.57.0
55
timm~=1.0.0
66
#accelerate>=0.32,<1.0.0

0 commit comments

Comments
 (0)