Phramer_AI / requirements.txt
Malaji71's picture
Update requirements.txt
5c7b1d3 verified
raw
history blame
505 Bytes
--extra-index-url https://download.pytorch.org/whl/cu121
gradio==5.33.1
torch==2.1.0
torchvision==0.16.0+cu121
transformers==4.44.0
accelerate==0.24.1
pillow==10.0.1
requests==2.31.0
numpy==1.26.4
sentence-transformers==2.2.2
einops==0.8.0
timm==1.0.3
opencv-python==4.8.1.78
opencv-contrib-python==4.8.1.78
deepface==0.0.79
tf-keras==2.15.0
mediapipe==0.10.7
retina-face==0.0.13
gdown==4.7.1
tqdm==4.66.1
# Flash Attention - use prebuilt wheel if available
flash-attn==2.7.4.post1 --no-build-isolation