From 09212bd8b08eb6453695b185de7c3683b78754e2 Mon Sep 17 00:00:00 2001 From: ale Date: Fri, 29 Mar 2024 18:32:27 +0100 Subject: [PATCH] initial commit --- Dockerfile | 7 +++++++ README.md | 18 ++++++++++++++++++ models/.gitkeep | 0 3 files changed, 25 insertions(+) create mode 100644 Dockerfile create mode 100644 README.md create mode 100644 models/.gitkeep diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..2a12e16 --- /dev/null +++ b/Dockerfile @@ -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"] diff --git a/README.md b/README.md new file mode 100644 index 0000000..3107788 --- /dev/null +++ b/README.md @@ -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/ --quantize --task +``` + +## License + +MIT \ No newline at end of file diff --git a/models/.gitkeep b/models/.gitkeep new file mode 100644 index 0000000..e69de29