Update requirements.txt

requirements.txt

@@ -45,7 +45,9 @@ executing==2.2.1
 faiss-cpu==1.12.0
 fastapi==0.116.1
 fastjsonschema==2.21.2
-# fbgemm_gpu==1.3.0  # REMOVED - GPU only library
+# CPU-only mode can use this instead of fbgemm_gpu
+# Comment out fbgemm_gpu if you're running in CPU-only mode
+fbgemm_gpu==1.3.0
 filelock==3.19.1
 findspark==2.0.1
 flake8==7.3.0
@@ -135,7 +137,22 @@ notebook==7.4.5
 notebook_shim==0.2.4
 numba==0.61.2
 numpy==2.2.6
-# All nvidia-* packages removed - GPU only
+# NVIDIA CUDA libraries - not required in CPU-only mode
+# Comment these out if you're running in CPU-only mode or don't have NVIDIA drivers
+nvidia-cublas-cu12==12.8.4.1
+nvidia-cuda-cupti-cu12==12.8.90
+nvidia-cuda-nvrtc-cu12==12.8.93
+nvidia-cuda-runtime-cu12==12.8.90
+nvidia-cudnn-cu12==9.10.2.21
+nvidia-cufft-cu12==11.3.3.83
+nvidia-cufile-cu12==1.13.1.3
+nvidia-curand-cu12==10.3.9.90
+nvidia-cusolver-cu12==11.7.3.90
+nvidia-cusparse-cu12==12.5.8.93
+nvidia-cusparselt-cu12==0.7.1
+nvidia-nccl-cu12==2.27.3
+nvidia-nvjitlink-cu12==12.8.93
+nvidia-nvtx-cu12==12.8.90
 oci==2.160.0
 openai==1.107.1
 opentelemetry-api==1.37.0
@@ -220,17 +237,18 @@ threadpoolctl==3.6.0
 tinycss2==1.4.0
 tokenizers==0.22.0
 toml==0.10.2
-# Replace GPU torch with CPU version
+# PyTorch and related libraries
+# For CPU-only mode, you can use:
+# pip install torch torchvision torchrec --index-url https://download.pytorch.org/whl/cpu
 torch==2.8.0
 torchmetrics==1.0.3
-torchrec==0.6.0  
-#fbgemm-gpu==0.6.0
+torchrec==0.6.0
 torchvision==0.23.0
 tornado==6.5.2
 tqdm==4.67.1
 traitlets==5.14.3
 transformers==4.56.1
-# triton==3.4.0  # REMOVED - GPU only
+triton==3.4.0
 types-python-dateutil==2.9.0.20250822
 typing-inspect==0.9.0
 typing-inspection==0.4.1
@@ -249,4 +267,4 @@ Werkzeug==3.1.3
 widgetsnbextension==4.0.14
 xxhash==3.5.0
 zipp==3.23.0
-zstd==1.5.7.2
+zstd==1.5.7.2