Browse Source

Add transformers version pinning for qwenvl2

tags/v0.3.7rc0
haixuanTao 1 year ago
parent
commit
0684f05e4e
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      node-hub/dora-qwenvl/pyproject.toml

+ 1
- 1
node-hub/dora-qwenvl/pyproject.toml View File

@@ -16,7 +16,7 @@ dora-rs = "^0.3.6"
numpy = "< 2.0.0"
torch = "^2.4.0"
torchvision = "^0.19"
transformers = { git = "https://github.com/huggingface/transformers" }
transformers = "^4.45"
qwen-vl-utils = "^0.0.2"
accelerate = "^0.33"
# flash_attn = "^2.6.1" # Install using: pip install -U flash-attn --no-build-isolation


Loading…
Cancel
Save