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 99 SingletonModelExplorerServer ,
1010 visualize ,
1111 visualize_graph ,
12+ visualize_model_explorer ,
1213)
Original file line number Diff line number Diff line change @@ -108,7 +108,7 @@ def visualize(
108108 ** kwargs ,
109109):
110110 """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
112112 from EdgeProgramManager and ExecutorchProgramManager for you.
113113
114114 See https://github.com/google-ai-edge/model-explorer/wiki/4.-API-Guide#visualize-pytorch-models
@@ -123,13 +123,22 @@ def visualize(
123123 )
124124 if reuse_server :
125125 cur_config .set_reuse_server ()
126- visualize_from_config (
127- cur_config ,
126+ visualize_model_explorer (
127+ config = kwargs . pop ( "config" , cur_config ) ,
128128 no_open_in_browser = no_open_in_browser ,
129129 ** kwargs ,
130130 )
131131
132132
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+
133142def visualize_graph (
134143 graph_module : GraphModule ,
135144 exported_program : ExportedProgram | EdgeProgramManager | ExecutorchProgramManager ,
You can’t perform that action at this time.
0 commit comments