Back to Chattts

Export onnx or JIT models for deployment

examples/onnx/README.md

0.2.5293 B
Original Source

Export onnx or JIT models for deployment

Run pip install onnx -U.

Export GPT

  1. Run python examples/onnx/exporter.py --gpt

Export other models

Run python examples/onnx/exporter.py --decoder --vocos

Reference

Run LLMs on Sophon TPU