mirror of
https://github.com/khoj-ai/khoj.git
synced 2024-11-23 15:38:55 +01:00
Pin pytorch version to 2.0.1 in order to avoid exit code 139 in Docker container (#512)
This commit is contained in:
parent
e3f8a95784
commit
6dc0df3afb
2 changed files with 3 additions and 1 deletions
|
@ -25,9 +25,11 @@ services:
|
|||
- ./tests/data/embeddings/:/root/.khoj/content/
|
||||
- ./tests/data/models/:/root/.khoj/search/
|
||||
- khoj_config:/root/.khoj/
|
||||
- sentence_tranformer_models:/root/.cache/torch/sentence_transformers
|
||||
# Use 0.0.0.0 to explicitly set the host ip for the service on the container. https://pythonspeed.com/articles/docker-connection-refused/
|
||||
command: --host="0.0.0.0" --port=42110 -vv
|
||||
|
||||
|
||||
volumes:
|
||||
khoj_config:
|
||||
sentence_tranformer_models:
|
||||
|
|
|
@ -52,7 +52,7 @@ dependencies = [
|
|||
"schedule == 1.1.0",
|
||||
"sentence-transformers == 2.2.2",
|
||||
"transformers >= 4.28.0",
|
||||
"torch >= 2.0.1",
|
||||
"torch == 2.0.1",
|
||||
"uvicorn == 0.17.6",
|
||||
"aiohttp == 3.8.5",
|
||||
"langchain >= 0.0.187",
|
||||
|
|
Loading…
Reference in a new issue