diff --git a/requirements/rocm.txt b/requirements/rocm.txt index 4ca7073830..5179f6ee8d 100644 --- a/requirements/rocm.txt +++ b/requirements/rocm.txt @@ -23,6 +23,7 @@ timm>=1.0.17 # To be consistent with test_quark.py amd-quark>=0.8.99 tilelang==0.1.10 - +# Required apache-tvm-ffi matching tilelang version +apache-tvm-ffi==0.1.10 # Required for faster safetensors model loading fastsafetensors >= 0.3.2 diff --git a/requirements/test/rocm.txt b/requirements/test/rocm.txt index eced711711..ce18ce456c 100644 --- a/requirements/test/rocm.txt +++ b/requirements/test/rocm.txt @@ -44,6 +44,7 @@ anyio==4.13.0 # watchfiles apache-tvm-ffi==0.1.10 # via + # -c requirements/rocm.txt # tilelang # xgrammar arctic-inference==0.1.1