Skip to content

GitLab

Commit 8680bfdb authored by zhuwenwen's avatar zhuwenwen
Browse files

[Deps] add torchvision

parent df4818c6
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
requirements/rocm.txt
View file @ 8680bfdb
......@@ -25,4 +25,5 @@ amd-quark>=0.8.99
# Other necessary dependencies
torch == 2.10.0
triton == 3.5.1
torchvision == 0.25.0
flash_attn == 2.8.3
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment