Spaces:
Paused
Paused
Commit
·
ae59a11
1
Parent(s):
a4f2f70
Update Dockerfile
Browse files- 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=
|
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
|