Abhishek commited on
Commit
84d9dd6
·
1 Parent(s): 63c03d2

chore: update model paths to point to /data/

Browse files
Dockerfile CHANGED
@@ -33,7 +33,7 @@ COPY --chown=user . /app
33
  ARG HF_TOKEN
34
  ENV HF_TOKEN=${HF_TOKEN}
35
  RUN pip install huggingface_hub hf_transfer
36
- RUN python scripts/download_gguf.py openbmb/MiniCPM5-1B-GGUF MiniCPM5-1B-Q4_K_M.gguf models/MiniCPM5-1B-Q4_K_M.gguf
37
 
38
  EXPOSE 7860 8080
39
  CMD ["python", "app.py"]
 
33
  ARG HF_TOKEN
34
  ENV HF_TOKEN=${HF_TOKEN}
35
  RUN pip install huggingface_hub hf_transfer
36
+ RUN python scripts/download_gguf.py openbmb/MiniCPM5-1B-GGUF MiniCPM5-1B-Q4_K_M.gguf /data/MiniCPM5-1B-Q4_K_M.gguf
37
 
38
  EXPOSE 7860 8080
39
  CMD ["python", "app.py"]
src/apps/p1_elder_paperwork/pipeline.py CHANGED
@@ -175,6 +175,7 @@ def get_llm():
175
  if not model_path:
176
  # Fallback to standard locations
177
  fallback_paths = [
 
178
  "models/MiniCPM5-1B-Q4_K_M.gguf",
179
  "/app/models/MiniCPM5-1B-Q4_K_M.gguf",
180
  "/root/docker-data/models/MiniCPM5-1B-Q4_K_M.gguf"
 
175
  if not model_path:
176
  # Fallback to standard locations
177
  fallback_paths = [
178
+ "/data/MiniCPM5-1B-Q4_K_M.gguf",
179
  "models/MiniCPM5-1B-Q4_K_M.gguf",
180
  "/app/models/MiniCPM5-1B-Q4_K_M.gguf",
181
  "/root/docker-data/models/MiniCPM5-1B-Q4_K_M.gguf"