caption: add ninja dep as it speeds up flash_attn build: better multiprocessing

Aaron Boxer · Aaron Boxer · commit 496ed714b719 · 2025-01-10T07:06:52.000-05:00
diff --git a/requirements.txt b/requirements.txt
@@ -6,8 +6,6 @@ torchvision
 torchaudio
 transformers
 numpy
-black
-ruff
 huggingface-hub
 ultralytics
 skia-python
diff --git a/requirements/caption_requirements.txt b/requirements/caption_requirements.txt
@@ -3,5 +3,6 @@
 
 # Note: building flash_attn also requires installation of nvidia-cuda-toolkit ubuntu package
 
+ninja
 flash_attn
 

Original file line number	Diff line number	Diff line change
`@@ -3,5 +3,6 @@`
`3`	`3`
`4`	`4`	`# Note: building flash_attn also requires installation of nvidia-cuda-toolkit ubuntu package`
`5`	`5`
	`6`	`+ninja`
`6`	`7`	`flash_attn`
`7`	`8`