diff --git a/requirements/rocm-test.txt b/requirements/rocm-test.txt index ee927accf..c29b67b7c 100644 --- a/requirements/rocm-test.txt +++ b/requirements/rocm-test.txt @@ -86,3 +86,5 @@ arctic-inference == 0.1.1 open-clip-torch==2.32.0 # Required for isaac Multi-Modal generation test perceptron==0.1.4 +# Required for the multi-modal models test +timm==1.0.17