Update Dockerfile

Dockerfile

@@ -1,19 +1,15 @@
 FROM ollama/ollama:latest
-
 USER root
 RUN apt-get update && apt-get install -y \
     python3 \
     python3-pip \
     && rm -rf /var/lib/apt/lists/*
-
 RUN mkdir -p /workspace/ollama \
              /workspace/webui \
              /workspace/home && \
     chmod -R 777 /workspace
-
 RUN pip install open-webui --break-system-packages && \
     chmod -R 777 /usr/local/lib/python3.12/dist-packages/open_webui/static
-
 USER 1000
 ENV HOME=/workspace/home \
     PATH=/usr/local/bin:$PATH \
@@ -21,9 +17,11 @@ ENV HOME=/workspace/home \
     OLLAMA_MODELS=/workspace/ollama \
     DATA_DIR=/workspace/webui \
     WEBUI_SECRET_KEY=supersecretkey123 \
-    HF_TOKEN=${HF_TOKEN}
-
+    HF_TOKEN=${HF_TOKEN} \
+    OLLAMA_FLASH_ATTENTION=1 \
+    OLLAMA_KV_CACHE_TYPE=q8_0 \
+    OLLAMA_NUM_PARALLEL=1 \
+    OLLAMA_CONTEXT_LENGTH=4096
 EXPOSE 7860
-
 ENTRYPOINT []
 CMD ["sh", "-c", "ollama serve & sleep 5 && open-webui serve --port 7860 --host 0.0.0.0"]