initial commit

This commit is contained in:
ale 2024-03-29 18:32:27 +01:00
commit 09212bd8b0
3 changed files with 25 additions and 0 deletions

7
Dockerfile Normal file
View 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
View 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
View File