neee

Dockerfile

@@ -14,6 +14,9 @@ RUN pip install --no-cache-dir --upgrade pip
 RUN pip install --no-cache-dir -r requirements.txt
 RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
 
+# ✅ Install accelerate (fixes distilgpt2 loading with device_map)
+RUN pip install --no-cache-dir accelerate
+
 # ✅ Set environment variables to a writable path
 ENV HF_HOME=/app/cache
 ENV TRANSFORMERS_CACHE=/app/cache
@@ -38,4 +41,4 @@ RUN chmod -R 777 /app/cache
 EXPOSE 7860
 
 # ✅ Run app
-CMD ["gunicorn", "--bind", "0.0.0.0:7860", "--workers", "1", "--timeout", "120", "app:app"]
+CMD ["gunicorn", "--bind", "0.0.0.0:7860", "--workers", "1", "--timeout", "120", "app:app"]