Update requirements.txt

requirements.txt

@@ -1,7 +1,26 @@
+# Transformers library by Hugging Face for LLMs
+transformers>=4.40.0  # Required for LLaMA 3.1 and PEFT support
 
-transformers
-accelerate
-bitsandbytes
-peft
-gradio
-datasets
+# Hugging Face Hub integration
+huggingface_hub>=0.21.4  # To download private models and access secrets via token
+
+# Gradio for UI in Hugging Face Spaces
+gradio>=4.19  # Provides web-based interface to interact with the assistant
+
+# BitsAndBytes for 4-bit and 8-bit quantized model loading (LLaMA 3.1 needs this for memory-efficient inference)
+bitsandbytes>=0.42.0  # Required for loading large models in 4-bit/8-bit
+
+# Accelerate for device mapping and optimized model training/inference
+accelerate>=0.27.0  # Makes it easy to run across CPU/GPU with minimal changes
+
+# PEFT (Parameter-Efficient Fine-Tuning) for QLoRA / LoRA-based fine-tuning
+peft>=0.10.0  # Used to fine-tune large language models efficiently
+
+# Datasets library to load and process your JSON dataset
+datasets>=2.19.0  # Useful for loading dataset.json and mapping instructions/responses
+
+# Tokenizers backend (implicitly used by transformers, included for stability)
+tokenizers>=0.19.1
+
+# Optional utilities (used for progress bars, etc.)
+tqdm>=4.66.0