Spaces:
Running
Running
Oleg Shulyakov
commited on
Commit
·
5b4b478
1
Parent(s):
05d1b68
Rename CUDA parameter
Browse files- docker-compose.yml +1 -3
- start.sh +1 -1
docker-compose.yml
CHANGED
|
@@ -1,5 +1,3 @@
|
|
| 1 |
-
# Docker compose file to LOCAL development
|
| 2 |
-
|
| 3 |
services:
|
| 4 |
gguf-my-repo:
|
| 5 |
build:
|
|
@@ -12,5 +10,5 @@ services:
|
|
| 12 |
volumes:
|
| 13 |
- .:/home/user/app
|
| 14 |
environment:
|
| 15 |
-
-
|
| 16 |
- HF_TOKEN=${HF_TOKEN}
|
|
|
|
|
|
|
|
|
|
| 1 |
services:
|
| 2 |
gguf-my-repo:
|
| 3 |
build:
|
|
|
|
| 10 |
volumes:
|
| 11 |
- .:/home/user/app
|
| 12 |
environment:
|
| 13 |
+
- RUN_CUDA=0
|
| 14 |
- HF_TOKEN=${HF_TOKEN}
|
start.sh
CHANGED
|
@@ -6,7 +6,7 @@ if [ ! -d "llama.cpp" ]; then
|
|
| 6 |
fi
|
| 7 |
|
| 8 |
export GGML_CUDA=OFF
|
| 9 |
-
if [[ -z "${
|
| 10 |
# enable CUDA if NOT running locally
|
| 11 |
export GGML_CUDA=ON
|
| 12 |
fi
|
|
|
|
| 6 |
fi
|
| 7 |
|
| 8 |
export GGML_CUDA=OFF
|
| 9 |
+
if [[ -z "${RUN_CUDA}" ]]; then
|
| 10 |
# enable CUDA if NOT running locally
|
| 11 |
export GGML_CUDA=ON
|
| 12 |
fi
|