Update Dockerfile with optimized settings for Kimi-K2

Dockerfile

@@ -6,17 +6,28 @@ WORKDIR /app
 RUN apt-get update && apt-get install -y \
     git \
     git-lfs \
+    build-essential \
     && rm -rf /var/lib/apt/lists/*
 
+# Initialize git-lfs
+RUN git lfs install
+
 # Install Python packages
 COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install --no-cache-dir --upgrade pip && \
+    pip install --no-cache-dir -r requirements.txt
 
 # Copy application files
 COPY . .
 
 # Set up Hugging Face cache directory
 ENV HF_HOME=/app/cache
+ENV TRANSFORMERS_CACHE=/app/cache
+ENV HF_HUB_CACHE=/app/cache
+
+# Set environment variables for better performance
+ENV PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:512
+ENV CUDA_VISIBLE_DEVICES=0,1,2,3
 
 # Expose port for the application
 EXPOSE 7860