Flux_Prompt_Optimizer / requirements.txt
Malaji71's picture
Update requirements.txt
8723d89 verified
raw
history blame
526 Bytes
# BAGEL 7B dependencies - Exact from working implementation
decord==0.6.0
einops==0.8.1
huggingface_hub==0.29.1
matplotlib==3.7.0
numpy==1.24.4
opencv_python==4.7.0.72
pyarrow==11.0.0
PyYAML==6.0.2
requests==2.32.3
safetensors==0.4.5
scipy==1.10.1
sentencepiece==0.1.99
torch==2.5.1
torchvision==0.20.1
transformers==4.49.0
accelerate>=0.34.0
wandb
# Gradio UI framework
gradio==5.33.2
# Hugging Face Spaces support
spaces==0.37.0
# Image processing
pillow==10.0.1
# Flash attention will be installed dynamically in app.py