Update requirements.txt with comprehensive dependencies for Kimi model

requirements.txt

@@ -1,12 +1,60 @@
-torch>=2.1.0
-transformers>=4.40.0
-peft>=0.10.0
-accelerate>=0.27.0
-bitsandbytes>=0.42.0
-gradio>=4.19.0
-huggingface-hub>=0.20.0
-sentencepiece>=0.1.99
-protobuf>=3.20.0
-safetensors>=0.4.0
-scipy>=1.10.0
+# Core ML Libraries
+transformers>=4.56.0          # Required by Kimi model (has assertion check)
+accelerate>=0.34.2            # Compatible with latest transformers
+peft>=0.13.2                  # Latest stable
+bitsandbytes>=0.45.1          # Compatible with triton 3.x (0.45.0 has triton.ops issue) 
+sentencepiece==0.2.0
+protobuf==5.29.2              # Updated for compatibility
 
+# Training & Optimization
+deepspeed==0.16.3             # Compatible with torch 2.5, more stable than 0.18.x
+triton>=3.2.0                 # Required by fla for optimal performance
+scipy==1.14.1                 # Updated
+scikit-learn==1.6.0           # Updated
+ninja==1.11.1.1
+
+# Data Processing
+datasets>=3.2.0               # Updated for compatibility
+tokenizers>=0.21.0            # Compatible with latest transformers
+pandas==2.2.3
+numpy==1.26.4                 # Keep for stability (2.x has breaking changes)
+
+# Monitoring & Logging
+wandb==0.19.1                 # Updated
+tensorboard==2.18.0
+tqdm==4.67.1                  # Updated
+psutil==6.1.1                 # Updated
+pynvml==11.5.3                # Updated
+
+# Evaluation
+rouge-score==0.1.2
+sacrebleu==2.4.3              # Updated
+bert-score==0.3.13
+
+# Utilities
+pyyaml==6.0.2
+python-dotenv==1.0.1
+huggingface-hub>=0.34.0       # Required by transformers >=4.56.0
+safetensors==0.4.5
+tiktoken==0.8.0               # Updated
+hf_transfer==0.1.8            # Updated
+
+# Kimi / Flash Linear Attention runtime (requires torch>=2.5)
+# Install from git to get latest version with fla.layers module
+git+https://github.com/sustcsonglin/flash-linear-attention.git@main
+
+# Required by Kimi tokenizer (tiktoken BPE loader)
+blobfile==3.0.0               # Updated
+
+# Web UI for HF Space
+gradio==4.44.1                # Web interface to keep Space alive
+
+# API (optional - not used with Gradio)
+# fastapi==0.115.6            
+# uvicorn[standard]==0.34.0   
+# python-multipart==0.0.20
+
+# Development
+pytest==8.3.4                 # Updated
+black==24.10.0                # Updated
+flake8==7.1.1