ChoruYt commited on
Commit
ccd75b1
·
verified ·
1 Parent(s): 17b1888

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -3
Dockerfile CHANGED
@@ -16,8 +16,8 @@ RUN printf 'void vkGetInstanceProcAddr(){}\nvoid vkCreateInstance(){}\nvoid vkDe
16
  # Create model directory
17
  RUN mkdir -p /app/models/gemma
18
 
19
- # --- KEY FIX: Use the new 'hf' command instead of 'huggingface-cli' ---
20
- RUN hf download litert-community/gemma-4-E2B-it-litert-lm gemma-4-E2B-it.litertlm --local-dir /app/models/gemma
21
 
22
  # Copy the server code
23
  COPY server.py .
@@ -30,7 +30,9 @@ USER user
30
  # Set environment variables
31
  ENV PORT=7860
32
  ENV LD_LIBRARY_PATH=/app
33
- ENV GEMMA_MODEL_PATH=/app/models/gemma/gemma-4-E2B-it.litertlm
 
 
34
 
35
  EXPOSE 7860
36
 
 
16
  # Create model directory
17
  RUN mkdir -p /app/models/gemma
18
 
19
+ # --- KEY FIX: Updated 'hf download' to fetch the 12B model ---
20
+ RUN hf download litert-community/gemma-4-12B-it-litert-lm gemma-4-12B-it.litertlm --local-dir /app/models/gemma
21
 
22
  # Copy the server code
23
  COPY server.py .
 
30
  # Set environment variables
31
  ENV PORT=7860
32
  ENV LD_LIBRARY_PATH=/app
33
+
34
+ # --- KEY FIX: Updated path to point to the 12B model file ---
35
+ ENV GEMMA_MODEL_PATH=/app/models/gemma/gemma-4-12B-it.litertlm
36
 
37
  EXPOSE 7860
38