From 4da685d978e912af3079ae43acc1be58338fd9f7 Mon Sep 17 00:00:00 2001 From: seruva19 Date: Tue, 24 Dec 2024 01:12:18 +0300 Subject: [PATCH] Update requirements.txt --- requirements.txt | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/requirements.txt b/requirements.txt index c7f3a06..ef4284e 100644 --- a/requirements.txt +++ b/requirements.txt @@ -11,6 +11,7 @@ tokenizers==0.20.1 accelerate==1.0.1 safetensors==0.4.5 opencv-python==4.8.0.74 +git+https://github.com/facebookresearch/pytorchvideo@ae9cfc6 gradio==3.50.2 toolz==1.0.0 psutil==5.9.5 @@ -19,5 +20,9 @@ torchsde==0.2.6 dynamicprompts==0.31.0 compel==2.0.3 huggingface-hub==0.25.2 -https://github.com/jllllll/bitsandbytes-windows-webui/releases/download/wheels/bitsandbytes-0.41.2.post2-py3-none-win_amd64.whl; platform_system == "Windows" -bitsandbytes==0.41.1; platform_system != "Windows" \ No newline at end of file +bitsandbytes==0.43.3 +https://github.com/woct0rdho/triton/releases/download/v3.0.0-windows.post1/triton-3.0.0-cp310-cp310-win_amd64.whl; platform_system == "Windows" +triton==3.0.0; platform_system != "Windows" +flash_attn==2.7.0; platform_system != "Windows" +#https://github.com/jllllll/bitsandbytes-windows-webui/releases/download/wheels/bitsandbytes-0.41.2.post2-py3-none-win_amd64.whl; platform_system == "Windows" +#bitsandbytes==0.41.1; platform_system != "Windows" \ No newline at end of file