File tree Expand file tree Collapse file tree 2 files changed +13
-3
lines changed Expand file tree Collapse file tree 2 files changed +13
-3
lines changed Original file line number Diff line number Diff line change 9
9
SingletonModelExplorerServer ,
10
10
visualize ,
11
11
visualize_graph ,
12
+ visualize_model_explorer ,
12
13
)
Original file line number Diff line number Diff line change @@ -108,7 +108,7 @@ def visualize(
108
108
** kwargs ,
109
109
):
110
110
"""Wraps the visualize_from_config call from model_explorer.
111
- For convenicence , figures out how to find the exported_program
111
+ For convenience , figures out how to find the exported_program
112
112
from EdgeProgramManager and ExecutorchProgramManager for you.
113
113
114
114
See https://github.com/google-ai-edge/model-explorer/wiki/4.-API-Guide#visualize-pytorch-models
@@ -123,13 +123,22 @@ def visualize(
123
123
)
124
124
if reuse_server :
125
125
cur_config .set_reuse_server ()
126
- visualize_from_config (
127
- cur_config ,
126
+ visualize_model_explorer (
127
+ config = kwargs . pop ( "config" , cur_config ) ,
128
128
no_open_in_browser = no_open_in_browser ,
129
129
** kwargs ,
130
130
)
131
131
132
132
133
+ def visualize_model_explorer (
134
+ ** kwargs ,
135
+ ):
136
+ """Wraps the visualize_from_config call from model_explorer."""
137
+ visualize_from_config (
138
+ ** kwargs ,
139
+ )
140
+
141
+
133
142
def visualize_graph (
134
143
graph_module : GraphModule ,
135
144
exported_program : ExportedProgram | EdgeProgramManager | ExecutorchProgramManager ,
You can’t perform that action at this time.
0 commit comments