diff --git a/requirements-linux.txt b/requirements-linux.txt index 697b912b..d4a8d62d 100644 --- a/requirements-linux.txt +++ b/requirements-linux.txt @@ -19,6 +19,7 @@ scenedetect[opencv] pytorch-lightning inquirer onnx +decord colored torch==2.3.1 torchvision==0.18.1 diff --git a/requirements-windows.txt b/requirements-windows.txt index 80319bd1..9cee5bf5 100644 --- a/requirements-windows.txt +++ b/requirements-windows.txt @@ -20,6 +20,7 @@ onnxruntime-directml pytorch-lightning inquirer onnx +decord colored tensorrt-cu12-bindings tensorrt-cu12_libs==10.0.1