wzebrowski commited on
Commit
ae59a11
·
1 Parent(s): a4f2f70

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -2,7 +2,7 @@
2
  # Initialize device type args
3
  # use build args in the docker build commmand with --build-arg="BUILDARG=true"
4
  ARG USE_CUDA=false
5
- ARG USE_OLLAMA=false
6
  # Tested with cu117 for CUDA 11 and cu121 for CUDA 12 (default)
7
  ARG USE_CUDA_VER=cu121
8
  # any sentence transformer model; models to use can be found at https://huggingface.co/models?library=sentence-transformers
 
2
  # Initialize device type args
3
  # use build args in the docker build commmand with --build-arg="BUILDARG=true"
4
  ARG USE_CUDA=false
5
+ ARG USE_OLLAMA=true
6
  # Tested with cu117 for CUDA 11 and cu121 for CUDA 12 (default)
7
  ARG USE_CUDA_VER=cu121
8
  # any sentence transformer model; models to use can be found at https://huggingface.co/models?library=sentence-transformers