export jacobi onnx python export_onnx_jacobi.py --model_path your_torch_path --guess_len 8 --generation_mode sample --device cuda