diff --git a/requirements.txt b/requirements.txt index 367628c..f122f0f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -7,6 +7,8 @@ opencv-python>=4.7.0 onnxruntime>=1.15.0 onnxruntime-gpu>=1.15.0 protobuf>=3.20.2,<6.0.0 +transformers>=4.30.0 +diffusers>=0.18.0 #SAM2 hydra-core>=1.3.0 @@ -17,5 +19,4 @@ iopath>=0.1.9 decord ftfy typing_extensions -triton-windows - +#triton-windows