Update requirements.txt

requirements.txt

@@ -3,35 +3,33 @@ numpy
 websockets
 aiohttp
 aiosignal
-async-timeout==4.0.3
-attrs==23.1.0
-charset-normalizer==3.2.0
-frozenlist==1.4.0
-multidict==6.0.4
-fastapi==0.100.1
-uvicorn==0.23.2
-jinja2==3.1.2
-humanize==4.7.0
-yarl==1.9.2
+async-timeout
+attrs
+charset-normalizer
+frozenlist
+multidict
+fastapi
+uvicorn
+jinja2
+humanize
+yarl
 python-multipart
+
 # AI/ML dependencies
-torch==2.0.1
-transformers==4.31.0
-scipy==1.11.2
-einops
-timm
-# Runtime dependencies
-pathlib==1.0.1
-psutil==5.9.5  # For system monitoring
-ujson==5.8.0   # Fast JSON processing
-tqdm==4.65.0   # Progress bars
-colorama==0.4.6  # Terminal colors
+torch
+transformers
 accelerate
+safetensors
+sentencepiece  # Required for Llama tokenizer
+bitsandbytes  # For 4-bit quantization support
+
+# Runtime dependencies
+psutil  # For system monitoring
+ujson  # Fast JSON processing
+tqdm  # Progress bars
+colorama  # Terminal colors
+
 # Development dependencies (not needed in production)
-mypy==1.5.1
-pytest==7.4.0
-black==23.7.0
-pylint==2.17.5
-uvicorn
-fastapi
-pillow
+pytest
+black
+pylint