Wang Qiang 6942144ad7 Stable Diffusion model checkpoint export to onnx. (#340) ...

* stable diffusion export onnx

* fix pre commit bugs

* fix bugs

* safety checker support

* test export stable diffusion

2023-06-28 13:26:19 +08:00

..

__init__.py

[to #42322933] Add exporter module for onnx,ts and other formats.

2022-09-21 14:25:06 +08:00

test_export_csanmt_model.py

Support csanmt exporting and refactor some code

2023-02-10 05:15:04 +00:00

test_export_face_detection_scrfd.py

support onnx export for SCRFD model

2023-03-01 10:02:49 +08:00

test_export_object_detection_damoyolo.py

feat: support onnx export for domain_specific_object_detection

2023-05-25 10:44:09 +08:00

test_export_sbert_sequence_classification.py

[to #47307896] update backbone-head

2023-02-10 06:46:47 +00:00

test_export_sbert_zero_shot_classification.py

Support csanmt exporting and refactor some code

2023-02-10 05:15:04 +00:00

test_export_speech_signal_process.py

[to #42322933] add ONNX exporter for ans dfsmn

2023-04-28 10:33:32 +08:00

test_export_stable_diffusion.py

Stable Diffusion model checkpoint export to onnx. (#340)

2023-06-28 13:26:19 +08:00

test_export_tf_model.py

Support csanmt exporting and refactor some code

2023-02-10 05:15:04 +00:00

test_export_token_classification.py

[to #42322933] add onnx/torchscript exporter for token classification models

2023-03-07 14:12:23 +08:00