initial commit
This commit is contained in:
commit
09212bd8b0
7
Dockerfile
Normal file
7
Dockerfile
Normal file
@ -0,0 +1,7 @@
|
|||||||
|
FROM debian:sid
|
||||||
|
RUN apt update && apt install -y git python3-pip && apt clean
|
||||||
|
RUN git clone --depth 1 https://github.com/xenova/transformers.js /transformers.js
|
||||||
|
WORKDIR /transformers.js/scripts
|
||||||
|
RUN pip3 install --break-system-packages -r requirements.txt
|
||||||
|
CMD "--help"
|
||||||
|
ENTRYPOINT ["python3", "convert.py"]
|
18
README.md
Normal file
18
README.md
Normal file
@ -0,0 +1,18 @@
|
|||||||
|
# docker-onnx-converted - convert AI models in [ONNX](https://onnx.ai/) format
|
||||||
|
|
||||||
|
## Build
|
||||||
|
|
||||||
|
```
|
||||||
|
docker buildx build -t docker-onnx-converter .
|
||||||
|
```
|
||||||
|
|
||||||
|
## Usage
|
||||||
|
|
||||||
|
```
|
||||||
|
docker run -ti --name docker-onnx-converter --rm -v $PWD/models:/models docker-onnx-converter \
|
||||||
|
--model_id /models/<path to model> --quantize --task <type of model task>
|
||||||
|
```
|
||||||
|
|
||||||
|
## License
|
||||||
|
|
||||||
|
MIT
|
0
models/.gitkeep
Normal file
0
models/.gitkeep
Normal file
Loading…
Reference in New Issue
Block a user