Skip to content

Commit 603489a

Browse files
committed
Update readme
1 parent 5835627 commit 603489a

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

README.md

+10-3
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,11 @@ use in distributed production environments.
66

77
## Running
88

9-
The easiest way to run this project is using docker
9+
The easiest way to run this project is using docker. There are two image families,
10+
one for cuda machines and one for cpu:
11+
12+
- Cuda: ghcr.io/aidotse/docling-inference:rev
13+
- CPU: ghcr.io/aidotse/docling-inference:cpu-rev
1014

1115
```bash
1216
# Create volumes to not have to download models every time
@@ -57,7 +61,7 @@ project. Follow their installation instructions if you do not have it.
5761
5862
```bash
5963
# Install the dependencies
60-
poetry install
64+
poetry install --with=<cuda or cpu>
6165

6266
# Activate the shell
6367
poetry shell
@@ -68,7 +72,10 @@ python src/main.py
6872

6973
## Building
7074

71-
Build the project docker image with `docker build -t docling-inference:dev .`
75+
Build the project docker image with one of the following commands
76+
77+
- Cuda: `docker build -t ghcr.io/aidotse/docling-inference:dev .`
78+
- CPU: `docker build -f Dockerfile.cpu -t ghcr.io/aidotse/docling-inference:dev .`
7279

7380
## Configuration
7481

0 commit comments

Comments
 (0)