From 79a9215b50901e756f601c6da0ad568c678f8289 Mon Sep 17 00:00:00 2001 From: CSTEZCAN <33690601+CSTEZCAN@users.noreply.github.com> Date: Sun, 2 Mar 2025 15:31:19 +0300 Subject: [PATCH] Update requirements.txt or create another requirements as requirements_after_torch.txt with flash_attn. --- requirements.txt | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/requirements.txt b/requirements.txt index 38e813d..db4ae03 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,10 +1,10 @@ torch==2.2.2 torchvision==0.17.2 -flash_attn-2.7.3+cu11torch2.2cxx11abiFALSE-cp311-cp311-linux_x86_64.whl timm==1.0.14 albumentations==2.0.4 -onnx==1.14.0 -onnxruntime==1.15.1 +#onnx provided versions not found +onnx +onnxruntime pycocotools==2.0.7 PyYAML==6.0.1 scipy==1.13.0 @@ -17,3 +17,5 @@ py-cpuinfo==9.0.0 huggingface-hub==0.23.2 safetensors==0.4.3 numpy==1.26.4 +#installation of flash_attn only works after proper installation of torch. +#flash_attn-2.7.3+cu11torch2.2cxx11abiFALSE-cp311-cp311-linux_x86_64.whl