diff --git a/requirements/cuda.txt b/requirements/cuda.txt index 06956415d072e..be6fe6c054394 100644 --- a/requirements/cuda.txt +++ b/requirements/cuda.txt @@ -12,4 +12,5 @@ torchvision==0.23.0 # Required for phi3v processor. See https://github.com/pytor # https://github.com/facebookresearch/xformers/releases/tag/v0.0.32.post1 xformers==0.0.32.post1; platform_system == 'Linux' and platform_machine == 'x86_64' # Requires PyTorch >= 2.8 # FlashInfer should be updated together with the Dockerfile -flashinfer-python==0.4.0 \ No newline at end of file +flashinfer-python==0.4.0 +apache-tvm-ffi==0.1.0b15