Update attentions_onnx.py

2025-04-04 03:38:58 +08:00 · 2024-11-24 15:18:09 +08:00 · 2024-11-24 15:18:09 +08:00 · f9ade11d77
commit f9ade11d77
parent c286480d10
1 changed files with 8 additions and 0 deletions
--- a/infer/lib/infer_pack/attentions_onnx.py
+++ b/infer/lib/infer_pack/attentions_onnx.py
@ -1,3 +1,11 @@
+############################## Warning! ##############################
+#                                                                    #
+#           Onnx Export Not Support All Of Non-Torch Types           #
+#           Include Python Built-in Types!!!!!!!!!!!!!!!!!           #
+#                   If You Want TO Change This File                  #
+#                  Do Not Use All Of Non-Torch Types!                #
+#                                                                    #
+############################## Warning! ##############################
 import copy
 import math
 from typing import Optional