Update requirements.txt

requirements.txt

@@ -1,17 +1,23 @@
 gradio>=4.25.0
 transformers>=4.38.0 
-# Using a specific version of transformers can sometimes help with compatibility.
-# You can also try without specifying version if you prefer latest.
-torch>=2.0.0 
-# Specify torch version, CPU version is usually fine for Spaces if no GPU selected.
+torch>=2.2.2
+# Using a slightly newer torch version, ensure it's compatible with bitsandbytes
 # For CPU-only: torch --index-url https://download.pytorch.org/whl/cpu
 
+# For Quantization
+bitsandbytes>=0.42.0 
+#pip install -U bitsandbytes
+
+# bitsandbytes is primarily for NVIDIA GPUs. CPU support is experimental or might require specific builds/compilations.
+# If you encounter issues with bitsandbytes on CPU, you might need to remove it and load the model without quantization,
+# or find a CPU-specific quantization library.
+
+accelerate>=0.25.0 
+# Accelerate is helpful for device_map="auto" and managing model loading
+
 # Libraries you found helpful for environment stability:
 pydantic==2.10.6
 google-genai==0.5.0 
-# As before, adjust google-genai version if you have a specific one that worked.
-# This isn't directly used by this app but kept for stability.
 python-dotenv>=0.15.0
 
-# sentencepiece is often a dependency for tokenizers, though not always explicitly needed
-# sentencepiece
+sentencepiece # Often needed by tokenizers