"git@developer.sourcefind.cn:zhaoyu6/sglang.git" did not exist on "988accbc1eb1a9f9a20087166e9c23f71cb8eb01"
Unverified Commit f5fb36e8 authored by dyning's avatar dyning Committed by GitHub
Browse files

Update program.py

parent 28ab6a3b
......@@ -191,8 +191,8 @@ def build_export(config, main_prog, startup_prog):
func_infor = config['Architecture']['function']
model = create_module(func_infor)(params=config)
image, outputs = model(mode='export')
fetches_var = sorted([outputs[name] for name in outputs])
fetches_var_name = [name for name in fetches_var]
fetches_var_name = sorted([name for name in outputs])
fetches_var = [outputs[name] for name in fetches_var_name]
feeded_var_names = [image.name]
target_vars = fetches_var
return feeded_var_names, target_vars, fetches_var_name
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment