# CI requirements - flash_attn excluded (requires CUDA compilation not available in GitHub Actions)
# Users installing locally will use requirements.txt which includes flash_attn

# Core dependencies for ComfyUI-Grounding

# For GroundingDINO, MM-GroundingDINO, and OWLv2 (HuggingFace transformers)
transformers>=4.30.0
timm>=0.9.0
einops>=0.6.0  # Required by Florence-2 and SA2VA model architectures

# For YOLO-World
ultralytics>=8.2.0
hf_transfer
peft

# Image processing (usually already installed with ComfyUI)
# pillow>=10.0.0

# Optional: supervision for advanced visualization
# supervision>=0.16.0

# For SAM2 segmentation nodes
huggingface_hub>=0.20.0
pyyaml>=6.0
tqdm>=4.65.0

# For flash attention (optional for SA2VA performance)
torch<=2.8.0
torchvision
torchaudio
# flash_attn - excluded in CI (requires CUDA compilation)
