feliksius commited on
Commit
6174412
·
verified ·
1 Parent(s): 5bb17ea

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -16,7 +16,8 @@ COPY . .
16
  # Set environment variables for Hugging Face cache
17
  ENV HF_HOME=/app/cache
18
  ENV TRANSFORMERS_CACHE=/app/cache
19
- ENV USE_QUANTIZATION=0 # Set to 1 if using quantization and GPU is available
 
20
 
21
  # Expose port for FastAPI
22
  EXPOSE 8000
 
16
  # Set environment variables for Hugging Face cache
17
  ENV HF_HOME=/app/cache
18
  ENV TRANSFORMERS_CACHE=/app/cache
19
+ # Set to 1 if using quantization and GPU is available
20
+ ENV USE_QUANTIZATION=0
21
 
22
  # Expose port for FastAPI
23
  EXPOSE 8000