export.py里面的export_pnnx函数import不了pnnx.wrapper

[WilliamGwok]

def export_pnnx(args, model):
import os.path as osp
import sys

import pnnx
from pnnx.wrapper import convert_inputshape_to_cmd

无法找到正确的pnnx版本

[WilliamGwok]

用的是2.0.0版本的model assistant

[baorepo]

@baorepo ➜ /workspaces/ModelAssistant (2.0.0) $ git diff 
diff --git a/tools/export.py b/tools/export.py
index bf22eff..1673021 100644
--- a/tools/export.py
+++ b/tools/export.py
@@ -286,12 +286,28 @@ def get_exported_file_name_from_precision(args, precision, ext: str = '') -> str
     )
 
 
+def convert_inputshape_to_cmd(input_shape):
+    """
+    Converts the input shape to a command-line argument string for pnnx.
+
+    Args:
+        input_shape (torch.Size or list): The input shape of the model.
+
+    Returns:
+        str: A command-line argument string representing the input shape.
+    """
+    if not isinstance(input_shape, (list, tuple)):
+        raise ValueError("Input shape must be a list or tuple")
+
+    cmd = "inputshape=" + ",".join(map(str, input_shape))
+    return cmd
+
+
 def export_pnnx(args, model):
     import os.path as osp
     import sys
 
     import pnnx
-    from pnnx.wrapper import convert_inputshape_to_cmd
 
     model.eval()
     pnnx_bin_path = osp.join(osp.dirname(pnnx.__file__), 'bin')

@WilliamGwok 试一下

[github-actions]

This issue is stale because it has been open 14 days with no activity. Remove stale label or comment or this will be closed in 7 days.

[github-actions]

This issue was closed because it has been stalled for 7 days with no activity.