requirements: refactor

aaron-boxer · aaron-boxer · commit 39f047d9a996 · 2025-01-09T14:44:19.000-05:00
diff --git a/requirements.txt b/requirements.txt
@@ -9,7 +9,6 @@ numpy
 black
 ruff
 huggingface-hub
-confluent-kafka
 ultralytics
 skia-python
 pycairo
diff --git a/requirements/caption_requirements.txt b/requirements/caption_requirements.txt
@@ -1,5 +1,7 @@
 # These are the requirements specific to the caption element i.e. phi3-vision model.
 # Flash attention takes a long time to build, so it is separated out into this file.
 
+# Note: building flash_attn also requires installation of nvidia-cuda-toolkit ubuntu package
+
 flash_attn
 
diff --git a/requirements/kafkasink_requirements.txt b/requirements/kafkasink_requirements.txt
@@ -0,0 +1 @@
+confluent-kafka
