Skip to content

Commit

Permalink
Add transformers version pinning for qwenvl2 (#665)
Browse files Browse the repository at this point in the history
Follwing
https://github.com/huggingface/transformers/releases/tag/v4.45.0 , this
will makes it possible to push `dora-qwenvl2` within pip and let people
use dora build to install qwenvl2.
  • Loading branch information
haixuanTao authored Oct 7, 2024
2 parents 156b08a + 7109226 commit 4e428ee
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion node-hub/dora-qwenvl/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 4e428ee

Please sign in to comment.